大家好,我是小魔龙,Unity3D软件工程师,VR、AR,虚拟仿真方向,不定时更新软件开发技巧,生活感悟,觉得有用记得一键三连哦。
“给定一颗二叉树,求出该树的节点个数。”
题目链接:
来源:力扣(LeetCode)
链接: 222. 完全二叉树的节点个数 - 力扣(LeetCode)
给你一棵 完全二叉树 的根节点 root ,求出该树的节点个数。
完全二叉树 的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层,则该层包含 1~ 2h 个节点。
示例 1:
输入: root = [1,2,3,4,5,6]
输出: 6
示例 2:
输入: root = []
输出: 0
题意要求给定一个完全二叉树,求出该数的节点个数。
对于任意二叉树,都可以通过广度优先搜索算法BFS计算节点个数,这道题给定的完全二叉树,可以使用完全二叉树的特点计算节点个数。
完全二叉树的特性可知,最左边的节点一定位于最底层,从根节点出发,每次访问左子节点,直到遇到叶子结点,该叶子结点即为完全二叉树的最左边的节点,经过的路径啊成都为最大层数h。
第i层包含2i个节点,最底层包含的节点数最少为1,最多为2h。
因此对于最大层数为h的完全二叉树,可以通过二分查找的方式得到完全二叉树的节点个数。
根据节点的个数范围判断得到节点个数k:
代码参考:
class Solution {
public int countNodes(TreeNode root) {
if (root == null) {
return 0;
}
int level = 0;
TreeNode node = root;
while (node.left != null) {
level++;
node = node.left;
}
int low = 1 << level, high = (1 << (level + 1)) - 1;
while (low < high) {
int mid = (high - low + 1) / 2 + low;
if (exists(root, level, mid)) {
low = mid;
} else {
high = mid - 1;
}
}
return low;
}
public boolean exists(TreeNode root, int level, int k) {
int bits = 1 << (level - 1);
TreeNode node = root;
while (node != null && bits > 0) {
if ((bits & k) == 0) {
node = node.left;
} else {
node = node.right;
}
bits >>= 1;
}
return node != null;
}
}
时间复杂度:O(log2 n)
其中n是完全二叉树的节点数。
空间复杂度:O(1)
只需要常数级的变量空间。
在判断第k个节点是否存在的时候,用到了位运算得到第k个节点对应的路径,判断路径是否存在,就可以判断第k个节点是否存在。
比如说第k个节点位于第h层,则k的二进制表示包含h+1位,其中最高位是1,其余各位从高到底表示根节点到第k个节点的路径。
0表示移动左子节点,1表示移动到右子节点: