一棵完全二叉树的节点总数

一棵完全二叉树的节点总数

题意：已知一棵完全二叉树，求其节点的个数
要求：时间复杂度低于O(N)，N为这棵树的节点个数。
公式节点个数=2^n-1

思路：

如果是一个完全二叉树最左边一定有节点求出二叉树的高度H

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

再求出二叉树的右节点的最左边的节点是否等于二叉树的高度H
if==H说明至少二叉树的左子树是一棵满二叉树 else 右子树是一颗满二叉树。
具体代码如下：

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));
	}

	public static int bs(Node node, int leval, int h) {
		if (leval == h) {
			return 1;
		}
		if (mostLeftLevel(node.right, leval + 1) == h) {
		//左子树是一棵满二叉树
			return (1 << (h - leval)) + bs(node.right, leval + 1, h);
		} else {
		//右子树是一颗满二叉树高度比树的高度减1
			return (1 << (h - leval - 1)) + bs(node.left, leval + 1, h);
		}
	}