给出一个完全二叉树,求出该树的节点个数。
说明:
完全二叉树的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层,则该层包含 1~ 2h 个节点。
示例:
输入:
1
/ \
2 3
/ \ /
4 5 6
输出: 6
这一题遍历二叉树容易超时,有一种巧妙的方法,因为它是完全二叉树,所以可以根据满二叉树的方式求节点数。满二叉树的节点数是 1 << deep - 1
, deep是深度。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public int countNodes(TreeNode root) {
if (root == null)
return 0;
int n = fullTreeNode(root);
if (n != 0)
return n;
else
return countNodes(root.left) + countNodes(root.right) + 1;
}
public int fullTreeNode(TreeNode tree) {
int l = 0;
TreeNode t = tree;
while (t != null) {
l++;
t = t.left;
}
int r = 0;
t = tree;
while (t != null) {
r++;
t = t.right;
}
if (l == r)
return (1 << l) - 1;
return 0;
}
}