左神视频day05——题目八：求完全二叉树的节点个数

已知一棵完全二叉树，求其节点的个数。要求：时间复杂度低于O(N)，N为这棵树的节点个数

• 1 << h 等价于 2的h次方
• 如果根节点右子树的左子树到达最后一层，则根节点的左子树是满二叉树，加上根节点，节点数为2的 h-level 次方
• 如果根节点右子树的左子树没到最后一层，则根节点的右子树是满二叉树，加上根节点，节点数为2的 h-level-1 次方

// 1 << h  等价于  2的h次方
public class Code_08_CompleteTreeNodeNumber {

    public static class Node {
        public int value;
        public Node left;
        public Node right;

        public Node(int data) {
            this.value = data;
        }
    }

    public static int nodeNum(Node head) {
        if (head == null) {
            return 0;
        }
        return bs(head, 1, mostLeftLevel(head, 1));
    }

    /**
     * level表示当前节点的层数，根节点是第一层，即根节点处level=1
     * h表示这整棵二叉树的深度
     * 如果根节点右子树的左子树到达最后一层，则根节点的左子树是满二叉树，加上根节点，节点数为2的 h-level 次方
     * 如果根节点右子树的左子树没到最后一层，则根节点的右子树是满二叉树，加上根节点，节点数为2的 h-level-1 次方
     */
    public static int bs(Node node, int level, int h) {
        if (level == h) {
            return 1;
        }
        if (mostLeftLevel(node.right, level + 1) == h) {
            return (1 << (h - level)) + bs(node.right, level + 1, h);
        } else {
            return (1 << (h - level - 1)) + bs(node.left, level + 1, h);
        }
    }

    public static int mostLeftLevel(Node node, int level) {
        while (node != null) {
            level++;
            node = node.left;
        }
        return level - 1;
    }

    public static void main(String[] args) {
        Node head = new Node(1);
        head.left = new Node(2);
        head.right = new Node(3);
        head.left.left = new Node(4);
        head.left.right = new Node(5);
        head.right.left = new Node(6);
        System.out.println(nodeNum(head));
    }
}