[leetcode]Count Complete Tree Nodes

原创 2015年11月20日 10:49:03

题目描述如下:

Given a complete binary tree, count the number of nodes.

Definition of a complete binary tree from Wikipedia:
In a complete binary tree every level, except possibly the last, is completely filled, and all nodes in the last level are as far left as possible. It can have between 1 and 2h nodes inclusive at the last level h.

因为树这样的题目也没办法很好的调试,所以之前也是CE了好多遍

第一版的代码如下:

public class Solution {
    public int res;
    Solution(){
        this.res = 0;
    }
    public void addCount(){
        this.res++;
    }
    public int getCount(){
        return this.res;
    }
    public int countNodes(TreeNode root) {
        if(root == null) return 0;
        addCount();
        if(root.left == null && root.right == null)
            return getCount();
        if(root.left != null){
            countNodes(root.left);
        }
        if(root.right != null)
            countNodes(root.right);
        return getCount();
    }
}

然后就TLE了。想想也是,毕竟是medium难度的题目,如果就这样暴力破解未免太过分了。

再读了一遍题目,完全二叉树左子树高度和右子树高度一致,这时候若树的高度为height,那么一共有height^2-1个节点,这个地方可以进行剪枝。所以,只要递归左右两边子树高度不同的部分就好。利用这个思路再写就AC了。附上代码:

public class Solution {
    public int res;
    Solution(){
        this.res = 0;
    }
    public int getLeftHeight(TreeNode root){
        int height = 0;  
        while(root.left != null) {  
            root = root.left;  
            height ++;  
        }  
        return height;  
    }
    public int getRightHeight(TreeNode root){
        int height = 0;  
        while(root.right != null) {  
            root = root.right;  
            height ++;  
        }  
        return height;  
    }
    public int countNodes(TreeNode root) {
        if(root == null) return 0;
        int lheight = getLeftHeight(root);
        int rheight = getRightHeight(root);
        if(lheight == rheight) return (2 << lheight) - 1;
        else return countNodes(root.left) + countNodes(root.right) + 1;
    }
}

题目链接:https://leetcode.com/problems/count-complete-tree-nodes/

版权声明:本文为博主原创文章,未经博主允许不得转载。

Leetcode[222]-Count Complete Tree Nodes

Given a complete binary tree, count the number of nodes.Definition of a complete binary tree from Wi...
  • Dream_angel_Z
  • Dream_angel_Z
  • 2015年06月11日 13:24
  • 1508

leetcode 222: Count Complete Tree Nodes

leetcode 222: Count Complete Tree Nodes c++ java python
  • xudli
  • xudli
  • 2015年06月06日 05:14
  • 14227

LeetCode222:Count Complete Tree Nodes

Given a complete binary tree, count the number of nodes. Definition of a complete binary tree from W...
  • u012501459
  • u012501459
  • 2015年07月29日 17:17
  • 836

LeetCode222 Count CompleteTree Nodes(计算完全二叉树的节点数) Java 题解

题目: Given a complete binary tree, count the number of nodes. Definition of a complete binary...
  • u012249528
  • u012249528
  • 2015年07月09日 20:51
  • 1216

Count Complete Tree Nodes -- leetcode

Given a complete binary tree, count the number of nodes. Definition of a complete binary tree from ...
  • elton_xiao
  • elton_xiao
  • 2015年09月10日 15:36
  • 275

[LeetCode]Count Complete Tree Nodes

Given a complete binary tree, count the number of nodes. Definition of a complete binary tree from ...
  • CiaoLiang
  • CiaoLiang
  • 2015年06月06日 20:57
  • 1130

[LeetCode] Count Complete Tree Nodes

Given a complete binary tree, count the number of nodes.Definition of a complete binary tree from Wi...
  • u011331383
  • u011331383
  • 2015年06月06日 13:45
  • 1417

Leetcode Count Complete Tree Nodes

Leetcode Count Complete Tree Nodes,本算法主要可以使用二分搜索加速计算出最后一层最右边缺少的结点数,通过二分法,计算量变为O(h * h),其中h为树高,相对递归等方...
  • u010842603
  • u010842603
  • 2015年12月08日 11:58
  • 279

leetcode Count Complete Tree Nodes

对于每个节点如果左子树的高度等于右子树的高度,则直接套公式2^h-1,否则等于左子树的结点数量+右子树的结点数量+1 public class Solution {     public int ...
  • eshaoliu
  • eshaoliu
  • 2015年06月16日 20:34
  • 289

【Leetcode】Count Complete Tree Nodes

题目链接:https://leetcode.com/problems/count-complete-tree-nodes/ 题目: Given a complete binary tree, co...
  • yeqiuzs
  • yeqiuzs
  • 2015年11月24日 15:19
  • 356
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:[leetcode]Count Complete Tree Nodes
举报原因:
原因补充:

(最多只允许输入30个字)