二分查找+位运算 计算完全二叉树的节点个数（C++）

222. 完全二叉树的节点个数

给你一棵 完全二叉树 的根节点 root ，求出该树的节点个数。
完全二叉树 的定义如下：在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层，则该层包含 1~ 2h 个节点。

示例 1：
输入：root = [1,2,3,4,5,6]
输出：6

示例 2：
输入：root = []
输出：0

示例 3：
输入：root = [1]
输出：1

解题思路：这道题一开始考虑是完全二叉树，想着只要知道有多少层，那么除了最后一层的节点数不知道，其余层的所有节点数就是2^depth -1。所以开始考虑最后一层的节点数怎么计算呢？

如果再用队列进行层次遍历，那知道完全二叉树的层次也就没有什么意义了，用递归也是同样的。

最终还是参考了leetcode的官方题解，发现它是用了位运算，完全二叉树的节点范围一定在[2^depth，2^depth+1 -1]之间，这里根节点那层为第0层。

然后我们可以判断节点k，k介于[2^depth，2^depth+1 -1]之间，k的二进制为，最高位是1，其余是根节点到k节点的路径，0表示左路径，1表示右路径。比如k=10，10的二进制为1010，最高位为1，010表示左右左，正好是从根节点到k节点的路径。

如下图所示，bits=2^depth-1，用bits&k，能得到根节点到k节点的第一段路径，然后bits>>1， bits&k，得到第二段路径，不断循环，直到bits=0或者node等于空为止，这样可以找到k节点，返回true，如果k节点不存在，就返回false。这样就能确定k节点存在不存在。

然后我们可以使用二分查找，先找[2^depth，2^depth+1 -1]区间的中间节点mid，其中low最开始等于2^depth，high等于2^depth+1 -1。如果中间节点存在，low=mid，不存在，high=mid-1，继续在[low,high]区间进行查找，不断循环，最终能找到完全二叉树的最后一个节点，此时low对应的就是最后一个节点，最后返回low。

代码：

class Solution {	//222. 完全二叉树的节点个数
public:
	bool exists(TreeNode* root, int depth, int k) {
		int bits = 1 << (depth - 1);
		TreeNode* node = root;
		while (node != nullptr && bits > 0) {
			if (!(bits & k)) {
				node = node->left;
			}
			else {
				node = node->right;
			}
			bits >>= 1;
		}
		return node != nullptr;
	}

	int countNodes(TreeNode* root) {
		if (root == nullptr) return 0;
		int depth = 0;
		TreeNode* cur = root;
		//计算层数 根节点为第0层
		while (cur->left) {	
			depth++;
			cur = cur->left;
		}
		//节点数量的范围一定在[2^depth, 2^(depth+1)-1]之间，通过位运算以及二分查找来确定
		int low = 1 << depth, high = (1 << (depth + 1)) - 1;
		while (low < high) {
			int mid = ((high - low + 1) >> 1) + low;
			if (exists(root, depth, mid)) {
				low = mid;
			}
			else {
				high = mid - 1;
			}
		}
		return low;
	}
};

参考资料：leetcode-完全二叉树的节点个数-官方题解