【数据结构】二叉树的构建与基本操作实现

简介: 【数据结构】二叉树的构建与基本操作实现

1.前序建立二叉树

首先我们输入一段前序序列,用字符串存储,空节点的部分我们用字符'#' 替代,利用*pi作为下标,遍历该字符串。

前序:将字符赋给当前节点的数据域,*pi的值+1,给当前节点的左孩子执行相同操作,给当前节点的右孩子执行相同操作,采用递归方式完成。

那么你可以写出中序建立二叉树的算法么?

代码实现:

BTNode* BinaryTreeCreate(BTDataType* a, int* pi)
{
  if (a[*pi] == '#')
  {
    (*pi)++;
    return NULL;
  }
  BTNode* root = (BTNode*)malloc(sizeof(BTNode));
  if (root == NULL)
  {
    perror("malloc fail");
    exit(-1);
  }
  root->_data = a[(*pi)++];
  root->_left = BinaryTreeCreate(a, pi);
  root->_right= BinaryTreeCreate(a, pi);
  return root;
}

为什么使用指针pi,而不用整型pi?

  • 若传整型,在函数递归的过程中,由于函数栈帧的关系,整型pi的值不会保留,所以需要传递指针。

2.销毁二叉树

销毁二叉树比较简单。

注意:销毁二叉树采用的是后序遍历的方式,因为如果先销毁根节点,那么就找不到孩子节点了。

代码实现:

// 二叉树销毁
void BinaryTreeDestory(BTNode** root)
{
  if (*root)
  {
    BinaryTreeDestory(&(*root)->_left);
    BinaryTreeDestory(&(*root)->_right);
    free(*root);
        *root = NULL;
  }
}

3.统计

我们主要看一下统计第k层节点的算法:

如何判断何时递归到第k层呢?

  • 想要到达第k层,我们可以利用递归每次让k-1,当k的值为1的时候,那么此时我们就来到了第k层
  • 可以来到第k层的,也就是k==1时我们返回1
  • 递归还是采用双路递归,即返回左右子树的和即可

代码实现:

//计算节点数
int BinaryTreeSize(BTNode* root)
{
  if (root == NULL)
  {
    return 0;
  }
  return BinaryTreeSize(root->_left) + BinaryTreeSize(root->_right) + 1;
}
//计算叶子节点数
int BinaryTreeLeafSize(BTNode* root)
{
  if (root == NULL)
  {
    return 0;
  }
  if (root->_left == NULL && root->_right == NULL)
  {
    return 1;
  }
  return BinaryTreeLeafSize(root->_left) 
        + BinaryTreeLeafSize(root->_right);
}
// 二叉树第k层节点个数(假设从第1层开始)
int BinaryTreeLevelKSize(BTNode* root, int k)
{
  assert(k > 0);
  if (root == NULL)
  {
    return 0;
  }
  if (k == 1)
  {
    return 1;
  }
  return BinaryTreeLevelKSize(root->_left, k-1) 
        + BinaryTreeLevelKSize(root->_right, k-1);
    //递归每次让k-1,当k的值为1的时候,那么此时我们就来到了第k层
}
//二叉树深度
int BinaryTreeHeight(BTNode* root)
{
  if (root == NULL)
  {
    return 0;
  }
  return fmax(BinaryTreeHeight(root->_left)
        , BinaryTreeHeight(root->_right))+1;
}

4.查找值为x的节点

该递归过程可以理解为前序思想,即如果根节点的值为x,那么肯定就可以直接返回,如果不同,我们就可以向左子树和右子树方向考虑,具体的设计可见代码。

代码实现:

BTNode* BinaryTreeFind(BTNode* root, BTDataType x)
{
  if (root == NULL)
  {
    return NULL;
  }
  if (root->_data == x)
  {
    return root;
  }
  BTNode* ret = NULL;
  ret = BinaryTreeFind(root->_left,x);
  if (ret)
  {
    return ret;
  }
  ret = BinaryTreeFind(root->_right, x);
  if (ret)
  {
    return ret;
  }
  return NULL;
}

5.前中后序遍历

代码实现:

// 二叉树前序遍历 
void BinaryTreePrevOrder(BTNode* root)
{
  if (root == NULL)
  {
    return;
  }
  printf("%c ", root->_data);
  BinaryTreeInOrder(root->_left);
  BinaryTreeInOrder(root->_right);
}
// 二叉树中序遍历
void BinaryTreeInOrder(BTNode* root)
{
  if (root == NULL)
  {
    return;
  }
  BinaryTreeInOrder(root->_left);
  printf("%c ", root->_data);
  BinaryTreeInOrder(root->_right);
}
// 二叉树后序遍历
void BinaryTreePostOrder(BTNode* root)
{
  if (root == NULL)
  {
    return;
  }
  BinaryTreeInOrder(root->_left);
  BinaryTreeInOrder(root->_right);
  printf("%c ", root->_data);
}

6.层序遍历

层序遍历的实现需要用到队列的逻辑结构。

首先将根节点入队,输出队首元素,并根据根节点找到左右孩子节点并入队,然后出队根节点,此时队首元素更新为根节点的左子树,再以此左子树为根循环这个过程,就能成功实现层序遍历。

代码实现:

void BinaryTreeLevelOrder(BTNode* root)
{
  Que q;
  QueueInit(&q);
  if (root)
    QueuePush(&q, root);
  while (!QueueEmpty(&q))
  {
    BTNode* front = QueueFront(&q);
    printf("%c ", front->_data);
    if (front->_left)
    {
      QueuePush(&q, front->_left);
    }
    if (front->_right)
    {
      QueuePush(&q, front->_right);
    }
    QueuePop(&q);
  }
  QueueDestroy(&q);
  return;
}

7.判断二叉树是否为完全二叉树

完全二叉树通俗的讲就是每个节点都是连续的,不存在某个节点之前存在空节点的情况,那么根据这个特性,我们同样可以利用层序遍历的思想,利用队列的逻辑结构来解决。


思路:与层序遍历不同的是,我们不管左右子树是否为空都入队,这样在循环结束时,队列中要么全为空,此时证明该二叉树是完全二叉树,如果队列中但凡存在一个不为空的情况,那就证明该二叉树不为完全二叉树。


代码实现:

bool BinaryTreeComplete(BTNode* root)
{
  Que q;
  QueueInit(&q);
  if (root)
    QueuePush(&q, root);
  while (!QueueEmpty(&q))
  {
    BTNode* front = QueueFront(&q);
    if (front == NULL)
    {
      break;
    }
    QueuePush(&q, front->_left);
    QueuePush(&q, front->_right);
    QueuePop(&q);
  }
  while (!QueueEmpty(&q))
  {
    BTNode* front = QueueFront(&q);
    QueuePop(&q);
    if (front != NULL)
    {
      QueueDestroy(&q);
      return false;
    }
  }
  QueueDestroy(&q);
  return true;
}

注意:循环条件中QueueEmpty判断的是队列是否为空,即队首指针指向是否为空,而第一次遍历队列判断的条件是队首元素的数据域是否为NULL,两者不一样。

总结

递归是一种十分巧妙的方法,他的特点是可以拆分子问题,将大问题简化为可以解决小问题,但不知道你是否和我一样,思考问题好像已经习惯了遍历枚举求解的思想,那么下一篇文章我会为大家带来二叉树的OJ题系列,强化对于递归问题的理解,让我们一起努力吧🌝


目录
相关文章
|
11天前
|
存储 算法 关系型数据库
数据结构与算法学习二一:多路查找树、二叉树与B树、2-3树、B+树、B*树。(本章为了解基本知识即可,不做代码学习)
这篇文章主要介绍了多路查找树的基本概念,包括二叉树的局限性、多叉树的优化、B树及其变体(如2-3树、B+树、B*树)的特点和应用,旨在帮助读者理解这些数据结构在文件系统和数据库系统中的重要性和效率。
13 0
数据结构与算法学习二一:多路查找树、二叉树与B树、2-3树、B+树、B*树。(本章为了解基本知识即可,不做代码学习)
|
11天前
|
存储 算法 搜索推荐
数据结构与算法学习十七:顺序储存二叉树、线索化二叉树
这篇文章主要介绍了顺序存储二叉树和线索化二叉树的概念、特点、实现方式以及应用场景。
14 0
数据结构与算法学习十七:顺序储存二叉树、线索化二叉树
|
13天前
|
Java
【用Java学习数据结构系列】震惊,二叉树原来是要这么学习的(二)
【用Java学习数据结构系列】震惊,二叉树原来是要这么学习的(二)
23 1
|
13天前
|
算法 Java C语言
【用Java学习数据结构系列】震惊,二叉树原来是要这么学习的(一)
【用Java学习数据结构系列】震惊,二叉树原来是要这么学习的(一)
20 1
|
15天前
|
存储
【数据结构】二叉树链式结构——感受递归的暴力美学
【数据结构】二叉树链式结构——感受递归的暴力美学
|
11天前
|
存储 算法
探索数据结构:分支的世界之二叉树与堆
探索数据结构:分支的世界之二叉树与堆
|
11天前
|
存储 算法
数据结构与算法学习十六:树的知识、二叉树、二叉树的遍历(前序、中序、后序、层次)、二叉树的查找(前序、中序、后序、层次)、二叉树的删除
这篇文章主要介绍了树和二叉树的基础知识,包括树的存储方式、二叉树的定义、遍历方法(前序、中序、后序、层次遍历),以及二叉树的查找和删除操作。
15 0
|
11天前
|
算法 程序员 索引
数据结构与算法学习七:栈、数组模拟栈、单链表模拟栈、栈应用实例 实现 综合计算器
栈的基本概念、应用场景以及如何使用数组和单链表模拟栈,并展示了如何利用栈和中缀表达式实现一个综合计算器。
16 1
数据结构与算法学习七:栈、数组模拟栈、单链表模拟栈、栈应用实例 实现 综合计算器
|
11天前
初步认识栈和队列
初步认识栈和队列
35 10
|
5天前
数据结构(栈与列队)
数据结构(栈与列队)
11 1