记录新手小白的做题过程。
题目:给你一棵 完全二叉树 的根节点 root ,求出该树的节点个数。
完全二叉树 的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层,则该层包含 1~ 2h 个节点。
思路:1.递归遍历
2 .层序遍历把所有的结点都走一遍,别的博客已记录层序遍历,在循环里面将操作代码变一下就好了
递归很好理解啊,如果结点是空就返回0,若不是就计算左右结点长度加自己的长度1.
class Solution {
public:
int countNodes(TreeNode* root) {
if(root==nullptr)
return 0;
return countNodes(root->left)+countNodes(root->right)+1;
}
};
官方解答:
官方似乎想教会我一些东西,来看看官方的方法:二分查找 + 位运算。
对于任意二叉树,都可以通过广度优先搜索或深度优先搜索计算节点个数,时间复杂度和空间复杂度都是 O(n)O(n),其中 nn 是二叉树的节点个数。这道题规定了给出的是完全二叉树,因此可以利用完全二叉树的特性计算节点个数。
规定根节点位于第 00 层,完全二叉树的最大层数为 hh。根据完全二叉树的特性可知,完全二叉树的最左边的节点一定位于最底层,因此从根节点出发,每次访问左子节点,直到遇到叶子节点,该叶子节点即为完全二叉树的最左边的节点,经过的路径长度即为最大层数 hh。
因此对于最大层数为 hh 的完全二叉树,节点个数一定在 [2^h,2^{h+1}-1][2
h
,2
h+1
−1] 的范围内,可以在该范围内通过二分查找的方式得到完全二叉树的节点个数。具体做法是,根据节点个数范围的上下界得到当前需要判断的节点个数 kk,如果第 kk 个节点存在,则节点个数一定大于或等于 kk,如果第 kk 个节点不存在,则节点个数一定小于 kk,由此可以将查找的范围缩小一半,直到得到节点个数。
如何判断第 kk 个节点是否存在呢?如果第 kk 个节点位于第 hh 层,则 kk 的二进制表示包含 h+1h+1 位,其中最高位是 11,其余各位从高到低表示从根节点到第 kk 个节点的路径,00 表示移动到左子节点,11 表示移动到右子节点。通过位运算得到第 kk 个节点对应的路径,判断该路径对应的节点是否存在,即可判断第 kk 个节点是否存在。
评论区里有位大佬给出了很好的解释:
1 h = 0 / \ 2 3 h = 1 / \ / 4 5 6 h = 2
现在这个树中的值都是节点的编号,最底下的一层的编号是[2^h ,2^h - 1],现在h = 2,也就是4, 5, 6, 7。 4, 5, 6, 7对应二进制分别为 100 101 110 111 不看最左边的1,从第二位开始,0表示向左,1表示向右,正好可以表示这个节点相对于根节点的位置。 比如4的 00 就表示从根节点 向左 再向左。6的 10 就表示从根节点 向右 再向左 那么想访问最后一层的节点就可以从节点的编号的二进制入手。从第二位开始的二进制位表示了最后一层的节点相对于根节点的位置。 那么就需要一个bits = 2^(h - 1) 上面例子中的bits就是2,对应二进制为010。这样就可以从第二位开始判断。(树三层高,需要向左或向右走两次才能到叶子) 比如看5这个节点存不存在,先通过位运算找到编号为5的节点相对于根节点的位置。010 & 101 发现第二位是0,说明从根节点开始,第一步向左走。 之后将bit右移一位,变成001。001 & 101 发现第三位是1,那么第二步向右走。 最后bit为0,说明已经找到编号为5的这个节点相对于根节点的位置,看这个节点是不是空,不是说明存在,exist返回真 编号为5的节点存在,说明总节点数量一定大于等于5。所以二分那里low = mid 再比如看7存不存在,010 & 111 第二位为1,第一部从根节点向右;001 & 111 第三位也为1,第二步继续向右。 然后判断当前节点是不是null,发现是null,exist返回假。 编号为7的节点不存在,说明总节点数量一定小于7。所以high = mid - 1
意思就是用层数的(n-1)次方(减一是建立在根结点为第一层的前提下的,若根结点为0层,不需要减),与一个位置的二进制进行“与”运算,层数一定是第一个为1,而后面全为0的树 ,而第一个不需要算,遍历除了第一个的位数,若是0,往左走若是1,往右走,若走到此处为空,就记0
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;
}
}