【数据结构】链式二叉树

目录

前言

1、前置准备

2、二叉树的遍历

2.1、前序遍历/先根遍历

2.2、中序遍历/中根遍历

2.3、后序遍历/后根遍历

2.4、层序遍历 

3、二叉树相关问题

3.1、计算二叉树节点的个数

3.2、计算二叉树叶子节点的个数

3.3、计算二叉树的高度/深度

3.4、计算二叉树第K层的节点个数(K >= 1)

3.5、查找某个值所在的节点

3.6、判断是否是完全二叉树

3.7、二叉树的销毁

4、完整代码


前言

在学习二叉树的基本操作前,需先要创建一棵二叉树,然后才能学习其相关的基本操作。由于现在大家对二叉树结构掌握可能还不够深入,为了降低大家学习成本,此处手动快速创建一棵简单的二叉树,快速进入二叉树操作学习,等二叉树结构了解的差不多时,我们反过头再来研究二叉树真正的创建方式。

1、前置准备

先手动创建一个链式二叉树:

注意:以下代码并不是创建二叉树的方式,真正创建二叉树方式后序详解重点讲解。

typedef int BTDataType;
typedef struct BinaryTreeNode
{
    BTDataType data;   //存放数据
    struct BinaryTreeNode* left;  //左子树
    struct BinaryTreeNode* right; //右子树
}BTNode;

BTNode* BuyNode(BTDataType x)
{
    BTNode* newnode = (BTNode*)malloc(sizeof(BTNode)); //开辟空间
    if(newnode == NULL)  //开辟失败直接结束程序
    {
        perror("malloc fail");
        exit(-1);
    }
    newnode->data = x;  
    newnode->left = newnode->right = NULL; //初始化为空
    return newnode;
}

BTNode* CreatBinaryTree()
{
    BTNode* node1 = BuyNode(1);
    BTNode* node2 = BuyNode(2);
    BTNode* node3 = BuyNode(3);
    BTNode* node4 = BuyNode(4);
    BTNode* node5 = BuyNode(5);
    BTNode* node6 = BuyNode(6);

    node1->left = node2;   //把二叉树结构连接起来
    node1->right = node4;
    node2->left = node3;
    node4->left = node5;
    node4->right = node6;

    return node1;
}

在学习链式二叉树的基本操作之前,我们先来回顾一下二叉树的基本概念:

二叉树是:

1、空树

2、非空树:是由 根节点,根节点的左子树、根节点的右子树 组成的

从概念中可以看出,二叉树定义是 递归式 的,因此后序基本操作中基本都是按照该概念实现的。

2、二叉树的遍历

二叉树的遍历采用的是 分治思想 ,即将一个难以直接解决的大问题,分割成一些规模较小的相同问题,以便各个击破,分而治之。

所以我们可以把整个二叉树的大问题,通过递归分成一个个节点的小问题逐个处理。

2.1、前序遍历/先根遍历

先处理 根节点 ,再处理 左子树 ,最后处理 右子树 的顺序。

使用递归方法实现:

void PreOrder(BTNode* root)
{
    if(root == NULL) //如果是空节点,就打印后直接返回
    {
        printf("NULL ");
        return;
    }
    printf("%d ", root->data); //先打印该根节点,再遍历左子树,最后遍历右子树
    PreOrder(root->left);
    PreOrder(root->right);
}

因为是首次讲解递归方法,所以为了方便大家理解,我画了一张递归展开图,以后的递归代码就不再重复画了:

2.2、中序遍历/中根遍历

先处理 左子树 ,再处理 根节点 ,最后处理 右子树 的顺序。

 递归方法实现:

void InOrder(BTNode* root)
{
    if(root == NULL) //如果是空节点,就打印后直接返回
    {
        printf("NULL ");
        return;
    }
    InOrder(root->left); //先遍历左子树,再打印该根节点,最后遍历右子树
    printf("%d ", root->data);
    InOrder(root->right);
}

2.3、后序遍历/后根遍历

先处理 左子树 ,再处理 右子树 ,最后处理 根节点 的顺序。

 递归方法实现:

void PostOrder(BTNode* root)
{
    if(root == NULL) //如果是空节点,就打印后直接返回
    {
        printf("NULL ");
        return;
    }
    PostOrder(root->left); //先遍历左子树,再遍历右子树,最后打印该根节点
    PostOrder(root->right);
    printf("%d ", root->data);
}

2.4、层序遍历 

从第一行开始,从左向右依次遍历。

层序遍历是借助队列来实现的,逻辑如下:

队列里面存放数据的 数据类型 为二叉树节点的 结构体 ,但是由于结构体拷贝起来比较大,所以我们存放 结构体的指针

先判断根节点是否为空,如果根节点不为空,就把根节点放入队列,如果根节点为空,就不把根节点放入队列。

再判断队列是否为空,如果队列不为空,就while循环以下操作:把队头数据取出,并把对头数据的 左孩子 右孩子 当作根节点重复上面根节点的判断,队列为空的时候停止循环,层序遍历结束。

层序遍历的规律总结一句话就是出上一层,带下一层

void LevelOrder(BTNode* root)
{
    Queue q;
    QueueInit(&q);
    if(root)
        QueuePush(&q, root);
    
    while(!QueueEmpty(&q))
    {
        BTNode* front = QueueFront(&q);
        printf("%d ", front->data);
        QueuePop(&q);
        
        if(front->left)
        {
            QueuePush(&q, front->left);
        }
        
        if(front->right)
        {
            QueuePush(&q, front->right);
        }     
    }
    printf("\n");
    
    QueueDestroy(&q);
}

3、二叉树相关问题

3.1、计算二叉树节点的个数

解决这个问题有以下几种思路:

  1. 先创建一个计算节点数量的变量 size ,然后直接遍历二叉树,如果二叉树节点不是空,就让 size++ ,二叉树结点是空,就直接返回。
  2. 使用分治思想,把大问题分解成小问题,逐层统计。

思路一:

先创建一个计算节点数量的变量 size ,然后直接遍历二叉树,如果二叉树节点不是空,就让 size++ ,二叉树结点是空,就直接返回。

实现代码如下:

int TreeSize(BTNode* root)
{
    if(root == NULL)
    return;

    int size == 0;
    size++;
    TreeSize(root->left);
    TreeSize(root->right);
    return size;
}

先说结论,这种方法是 不可以 的。

如果是在普通循环程序外定义一个变量 size ,在循环内使用 size++  是可以的,因为每次 size++ 都是对同一个 size 进行的。但是如果是在 递归程序 中定义了一个变量 size ,那么在递归调用的过程会建立很多 函数栈帧 (这方面内容我在之前的一篇文章《函数栈帧的创建与销毁中有过详细的讲解》),每一个函数栈帧里都会定义一个变量 size ,这样就会导致每一次的 size++ 都是对不同的 size 进行的。

那么可能还会有同学会想到把 size 定义成 静态变量,使变量 size 创建在 静态区 而不在 栈区,这样问题不久解决了吗?

int TreeSize(BTNode* root)
{
    if(root == NULL)
    return;

    static int size == 0;
    size++;
    TreeSize(root->left);
    TreeSize(root->right);
    return size;
}

这样使用 static 来定义静态变量之后,如果同学们去尝试一下,会发现在第一次计算二叉树的节点数量时,结果是正确的,但是在第二次及之后再去计算节点数量的话,会发现结果越来越大,每一次计算都比上一次多。有的同学会有疑问,上面所写的程序不是在每次递归调用的时候都把 size 初始化为零了吗?为什么还是会越加越大呢?

这是因为 静态变量 全局变量 的作用域都是整个程序,所以只有在第一次进入函数时才会在定义的同时进行初始化,而如果使用 size = 0;来直接把 size 赋值为 0 的话,又没有办法控制什么时候把 size 赋值为 0 。

所以如果非要使用这种思想来计算二叉树节点数量的话,就只有一种方法,就是直接定义一个 全局变量 size ,并且 TreeSize 函数不使用返回值:

int size = 0;
void TreeSize(BTNode* root)
{
    if(root == NULL)
        return;
    size++;
    TreeSize(root->left);
    TreeSize(root->right);
}

需要注意的是,每次调用 TreeSize 函数时,都要手动让 size = 0


思路二:

使用分治思想,把大问题分解成小问题,逐层统计。

统计该节点子树的所有节点数量,并把统计到的节点数量加一,返回给该节点的父节点。

实现代码:

int TreeSize2(BTNode* root)
{
    if(root == NULL)
        return 0;
    return TreeSize2(root->left) + TreeSize2(root->right) + 1;
}

使用很少量的代码就可以实现预期的功能,方便又快捷。


所以 思路二 是计算二叉树节点数量的最优解。

3.2、计算二叉树叶子节点的个数

与计算二叉树节点个数相似,只不过不需要加上节点本身(即不需要每次都加 1 )。

int TreeLeafSize(BTNode* root)
{
    if(root == NULL) //不要忘记判断根节点是否为空,否则可能会出现对空节点解引用出错的问题
        return 0;
    if(root->left == NULL && root->right == NULL) //判断是否是叶子节点
        return 1;
    return TreeLeafSize(root->left) + TreeLeafSize(root->right); //把叶子节点都加起来
}

3.3、计算二叉树的高度/深度

因为二叉树的高度是由高度最高的子树所决定的,所以需要返回最高子树的高度。

所以我们依然使用分治思想,计算该节点的 左子树 右子树 的高度,再相互比较,最后返回最高子树的高度 + 1。

需要注意的是,在求 左子树 右子树 的高度时,必须采用 后序遍历 的方法,因为在求高度时,必须先递归到最末端的叶子节点,再从 0 开始层层往回依次加 1 ,然后再选取最大的那个子树的高度。

通过以上思想,可以写出以下代码:

int TreeHight(BTNode* root)
{
    if(root == NULL)
        return 0;
    return TreeHight(root->left) > TreeHight(root->right) 
           ? TreeHight(root->left) + 1 : TreeHight(root->right) + 1;
}

这段代码在逻辑上是正确的,但是依然存在很大的问题,因为每次比较完左子树与右子树的高度之后,直接就把已经得到的结果扔掉了,在返回结果时还要重新再计算一次高度,造成了非常大的性能浪费,所以需要再定义两个变量来保存已经得到的结果。

最终实现代码如下:

int TreeHight(BTNode* root)
{
    if(root == NULL)
        return 0;
    int leftHight = TreeHight(root->left);
    int rightHight = TreeHight(root->right);
    return leftHight > rightHight ? leftHight + 1 : rightHight + 1;
}

3.4、计算二叉树第K层的节点个数(K >= 1)

主要思路:

对于第 一 层,需要计算的是 左子树第 K 层的节点个数 + 右子树第 K 层的节点个数

对于第 二 层,需要计算的是 左子树第 K - 1 层的节点个数 + 右子树第 K - 1 层的节点个数

对于第 三 层,需要计算的是 左子树第 K - 2 层的节点个数 + 右子树第 K - 2 层的节点个数

      ............

对于第 K 层,需要计算的是 左子树第 一 层的节点个数 + 右子树第 一 层的节点个数

根据以上思路,我们需要根据 K 的数值分为两种情况分析:

1、当 K == 1 时,直接返回第一层的节点个数:1

2、当 K > 1 时,第 K 层转换为该节点的子节点所在层数的第 K - 1 层。

实现代码:

int TreeKLevelSize(BTNode* root, int k)
{
    if(root == NULL)
        return 0;

    if(k == 1)
        return 1;

    return TreeKLevelSize(root->left, k - 1)
         + TreeKLevelSize(root->right, k - 1);
}

3.5、查找某个值所在的节点

思路:如果当前节点就是指定数值所在的节点,那么直接返回该节点,如果当前节点不是指定节点,那么就先找左子树,再找右子树,如果最终还是没有找到指定节点,返回空。

BTNode* TreeFind(BTNode* root, BTDataType x)
{
    if(root == NULL)  //如果根节点是空,直接返回空
        return NULL;

    if(root->data == x)  //找到指定节点直接返回
        return root;

    BTNode* ret1 = TreeFind(root->left, x); //创建变量接收返回值
    if(ret1)  //如果接收到的返回值不是空,就说明找到指定节点了,返回指定节点
        return ret1;

    BTNode* ret2 = TreeFind(root->right, x);
    if(ret2)
        return ret2;

    return NULL; //什么都没找到,返回空
}

3.6、判断是否是完全二叉树

判断一个二叉树是否是完全二叉树,是否可以通过计算二叉树节点的数量来判断呢?答案是 不可以 的,因为完全二叉树节点的数量是一个范围,如果二叉树的高度为 h ,那么二叉树节点数量就在 2^(h - 1) ~ 2^h - 1 之间。

所以我们通过层序遍历来判断,只不过这里所用的层序遍历与上面所讲过的层序遍历有所不同,在这里 二叉树的空节点也要被排进队列当中 ,逻辑如下:

因为层序遍历的逻辑是出上一层,带下一层,所以再根据完全二叉树的性质:最后一层可以不满,但是所有节点必须从左到右连续排列。所以当队头数据为空时,只需要判断队列内还有没有非空节点就可以了,如果队列内还存在非空节点,就是非完全二叉树,如果队列内全是空节点,就是完全二叉树。

实现代码:

int TreeComplete(BTNode* root)
{
    Queue q;
    QueueInit(&q);
    if(root)
        QueuePush(&q, root);
    
    while(!QueueEmpty(&q))
    {
        BTNode* front = QueueFront(&q);
        QueuePop(&q);
        if(front == NULL)
        {
            break;
        }
        else
        {
            QueuePush(front->left);
            QueuePush(front->right);
        }
    }
    while(!QueueEmpty(&q))
    {
        BTNode* front = QueueFront(&q);
        if(front)
        {
            QueueDestroy(&q);
            return false;
        }
        else
        {
            QueuePop(&q);
        }
    }
    QueueDestroy(&q);
    return true;
}

3.7、二叉树的销毁

我们使用 后序遍历 来销毁二叉树:从最后一层开始,先销毁左子树,再销毁右子树,最后释放根节点,如果遇到空节点直接返回。

实现代码:

void TreeDestroy(BTNode* root)
{
    if(root == NULL)
    {
        return;
    }
    TreeDestroy(root->left);
    TreeDestroy(root->right);
    free(root);
}

需要注意的是:因为我们传递的是一级指针,而形参的改变不会影响实参,所以整个二叉树的根节点不会被销毁,需要用户自己在外面手动把根节点置空。

4、完整代码

因为之前已经对队列有过较为详细的讲解,所以这里就不再把队列的代码放出来了

#include <stdio.h>
#include "Queue.h"


BTNode* BuyBTNode(BTDataType x)
{
	BTNode* newnode = (BTNode*)malloc(sizeof(BTNode));
	if (newnode == NULL)
	{
		perror("malloc fail");
		exit(-1);
	}
	newnode->data = x;
	newnode->left = NULL;
	newnode->right = NULL;
	return newnode;
}

void PrevOrder(BTNode* root)
{
	if (root == NULL)
	{
		printf("NULL ");
		return;
	}
	printf("%d ", root->data);
	PrevOrder(root->left);
	PrevOrder(root->right);
}

void InOrder(BTNode* root)
{
	if (root == NULL)
	{
		printf("NULL ");
		return;
	}
	InOrder(root->left);
	printf("%d ", root->data);
	InOrder(root->right);
}

void PostOrder(BTNode* root)
{
	if (root == NULL)
	{
		printf("NULL ");
		return;
	}
	PostOrder(root->left);
	PostOrder(root->right);
	printf("%d ", root->data);
}

//求二叉树节点个数
int size = 0;
void TreeSize(BTNode* root)
{
	if (root == NULL)
	{
		return;
	}
	size++;
	TreeSize(root->left);
	TreeSize(root->right);
}

//求叶子节点的个数
int TreeLeafSize(BTNode* root)
{
	if (root == NULL)
	{
		return 0;
	}
	if (root->left == NULL && root->right == NULL)
	{
		return 1;
	}
	return TreeLeafSize(root->left) + TreeLeafSize(root->right);

}

//求二叉树的高度
int TreeHeight(BTNode* root)
{
	if (root == NULL)
	{
		return 0;
	}
	int LeftHeight = TreeHeight(root->left);
	int RightHeight = TreeHeight(root->right);

	return LeftHeight > RightHeight ? LeftHeight + 1 : RightHeight + 1;
}


//求第k层节点的个数 k>=1
int TreeKLevelSize(BTNode* root, int k)
{
	if (root == NULL)
	{
		return 0;
	}
	if (k == 1)
	{
		return 1;
	 }

	return TreeKLevelSize(root->left, k - 1) + TreeKLevelSize(root->right, k - 1);

}


//查找指定数字所在的节点
BTNode* TreeFind(BTNode* root, int k)
{
	if (root == NULL)
	{
		return NULL;
	}
	if (root->data == k)
	{
		return root;
	}
	BTNode* left = TreeFind(root->left, k);
	if (left)
	{
		return left;
	}
	BTNode* right = TreeFind(root->right, k);
	if (right)
	{
		return right;
	}
	return NULL;
}

void LevelOrder(BTNode* root)
{
	Queue q;
	QueueInit(&q);
	if (root)
	QueuePush(&q, root);

	while (!QueueEmpty(&q))
	{
		BTNode* front = QueueFront(&q);
		printf("%d ", front->data);
		QueuePop(&q);

		if (front->left)
		{
			QueuePush(&q, front->left);
		}
		if (front->right)
		{
			QueuePush(&q, front->right);
		}
	}
	printf("\n");
	QueueDestroy(&q);
}


//检查是否是完全二叉树
bool TreeComplete(BTNode* root)
{
	Queue q;
	QueueInit(&q);
	if (root)
	{
		QueuePush(&q, root);
	}
	while (!QueueEmpty(&q))
	{
		BTNode* front = QueueFront(&q);
		QueuePop(&q);
		if (!front)
		{
			break;
		}
		QueuePush(&q, front->left);
		QueuePush(&q, front->right);
	}

	//检查出到NULL以后后面是不是全为NULL
	while (!QueueEmpty(&q))
	{
		if (QueueFront(&q))
		{
			QueueDestroy(&q);
			return false;
		}
		QueuePop(&q);
	}
	QueueDestroy(&q);
	return true;
}

//使用后续遍历
void BinaryTreeDestory(BTNode* root)
{
	if (!root)
	{
		return;
	}
	BinaryTreeDestory(root->left);
	BinaryTreeDestory(root->right);
	free(root);
	//root = NULL;
}


int main()
{
	BTNode* n1 = BuyBTNode(1);
	BTNode* n2 = BuyBTNode(2);
	BTNode* n3 = BuyBTNode(3);
	BTNode* n4 = BuyBTNode(4);
	BTNode* n5 = BuyBTNode(5);
	BTNode* n6 = BuyBTNode(6);
	BTNode* n7 = BuyBTNode(7);
	n1->left = n2;
	n1->right = n4;
	n2->left = n3;
	n2->right = n7;
	n4->left = n5;
	n4->right = n6;

	PrevOrder(n1);
	printf("\n");
	InOrder(n1);
	printf("\n");
	PostOrder(n1);
	printf("\n");
	size = 0; 
	TreeSize(n1);
	printf("TreeSize: %d\n", size);

	size = 0;
	TreeSize(n1);
	printf("TreeSize: %d\n", size);

	size = 0;
	TreeSize(n1);
	printf("TreeSize: %d\n", size);

	printf("%d\n", TreeLeafSize(n1));

	printf("%d\n",TreeHeight(n1));

	printf("%d\n", TreeKLevelSize(n1, 2));

	printf("%d\n", TreeFind(n1, 4)->data);

	LevelOrder(n1);
	printf("%d ", TreeComplete(n1));
	BinaryTreeDestory(n1);
	n1 = NULL;
	return 0;
}

对于链式二叉树的基本操作就讲解到这里,希望同学们多多支持,如果有不妥的地方欢迎大佬指正,谢谢!

  • 16
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 7
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

世间是否此山最高

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值