初阶数据结构 二叉树常用函数(二)

函数一 求二叉树第K层的节点个数

还是一样 我们假设 K就是等于一

如果说是一个空数的话就返回0

如果说有值的话就返回一个1就可以

假设这个这层既不为空 又不是第K层的话 那么就说明第K层肯定是子树下面

那么就说明是左右子树的第(K-1)层

那么只将它们相加并且返回它们的值就好了

核心代码表示如下

复制代码
//树的第k层个数
//树的第k层个数 = 左子树第k-1层个数+右子树第k-1层个数
int TreeLevle(BTNode* root, int k)
{
	if (root == NULL)
	{
		return 0;
	}
	if (k == 1)
	{
		return 1;
	}
	int leftk = TreeLevle(root->left, k - 1);
	int rightk = TreeLevle(root->right, k - 1);

	return leftk + rightk;
}

我们来看看结果怎么样

函数二求二叉树的深度

这里还是一样 我们先来看图

我们先来看第极限的情况

假如我们的本身就是一个空树的话 我们可以直接返回0

如果不是空树的话我们可以寻找我们的左子树和右子树中的较大值(一样大返回哪一个都可以)

将它们加一后返回就可以

核心代码如下

复制代码
//树的高度
//当前树的高度=左右子树大的那个+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;
}

还是一样我们来看看效果

深度是4确实没错

函数三 求某个值为X的节点

还是一样 我们首先考虑极限情况

假设值就在根上

那么我们直接返回根的位置就好了

否则的话我们就往左边右边子树遍历

我们来看看核心代码

复制代码
//二叉树查找值为x的结点
BTNode* TreeFind(BTNode* root, BTDateType x)
{
	if (root == NULL)
	{
		return NULL;
	}
	if (root->data == x)
	{
		return root;
	}
	
	BTNode* lret = TreeFind(root->left, x);
	if(lret)
	return lret;
	
	BTNode* rret = TreeFind(root->right, x);
	if(rret)
	return rret;

	return NULL;
}

我们发现这里可以找出来了

以上就是本篇博客的全部内容啦,如有错误请各位大佬不吝赐教,感谢留言

相关推荐
不语n1 天前
快速排序(Quick Sort)详解与图解
数据结构·算法·排序算法·快速排序·双指针排序
三萬Q1 天前
数据结构--并查集
数据结构
桦说编程1 天前
如何在Java中实现支持随机访问的固定窗口队列
java·数据结构·后端
电鱼智能的电小鱼1 天前
基于电鱼 ARM 工控机的AI视频智能分析方案:让传统监控变得更聪明
网络·arm开发·人工智能·嵌入式硬件·算法·音视频
初学者,亦行者1 天前
Rust性能优化:内存对齐与缓存友好实战
算法·rust
py有趣1 天前
LeetCode算法学习之杨辉三角
学习·算法·leetcode
小白菜又菜1 天前
Leetcode 3100. Water Bottles II
算法·leetcode·职场和发展
北诺南兮1 天前
大模型算法面试笔记——多头潜在注意力(MLA)
笔记·深度学习·算法
微知语1 天前
悬垂引用的攻防战:Rust 如何从根源杜绝内存访问灾难
开发语言·算法·rust