2020/11/24 222. 完全二叉树的节点个数

一、BFS层序遍历

今天的每日一题是求完全二叉树节点的个数，拿到题目我第一反应是可以利用双向队列进行二叉树的BFS层序遍历，每添加一次节点，可以使得计数值加1。代码如下，但是提交后发现运行时间较慢。

class Solution {
    public int countNodes(TreeNode root) {
        Deque<TreeNode> queue = new LinkedList<>();
        if(root==null)return 0;
        queue.offer(root);
        int count = 1;
        TreeNode tmp;
        while(!queue.isEmpty()){
            tmp = queue.poll();
            if(tmp.left!=null){
                queue.offer(tmp.left);
                count++;
            }
            if(tmp.right!=null){
                queue.offer(tmp.right);
                count++;
            }
        }
        return count;
    }
}

二、DFS深度优先搜索

当我们直接利用DFS进行深度优先搜索时，由于没有出队和入队操作，运行时间缩短。

class Solution {
    public int countNodes(TreeNode root) {
        if(root==null)return 0;
        return countNodes(root.left)+countNodes(root.right)+1;
    }
}

三、递归改进

接下来对代码进行改进，主要目的为了降低时间复杂度。可以看到由于是完全二叉树，所以节点个数只会在最后一层发生变化，底层上方所有节点个数都与层高有着对应的函数关系。
在学习了精选解答后，有一种新的解法，可以避免递归调用一半的树，时间复杂度为T(n) = T(n/2) + logn。
主要思想为对根结点左右子树高度进行计算，当左子树高度等于右子树时，此时左子树必定满树；同样的当二者高度不同时，右子树满树。

class Solution {
    public int countNodes(TreeNode root) {
        if(root==null)return 0;
        int left = countLevels(root.left);
        int right = countLevels(root.right);
        if(left==right){
            return countNodes(root.right)+(1<<left);
        }else{
            return countNodes(root.left)+(1<<right);
        }
    }
    public int countLevels(TreeNode root){
        int count = 0;
        while(root!=null){
            root = root.left;
            count++;
        }
        return count;
    }
}