LeetCode 222. Count Complete Tree Nodes

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了LeetCode 222. Count Complete Tree Nodes相关的知识,希望对你有一定的参考价值。

complete binary tree:除最后一行外每一行的节点都有两个儿子,最后一行的节点尽可能靠左。

ver0:

1 class Solution {
2 public:
3     int countNodes(TreeNode* root) {
4         if(!root) return 0;
5         return 1 + countNodes(root->left) + countNodes(root->right);
6     }
7 };

不出意料地TLE。

 

ver1:

 1 class Solution {
 2 public:
 3     int countNodes(TreeNode* root) {
 4         if(!root) return 0;
 5         TreeNode* l = root, * r = root;
 6         int llen = 0, rlen = 0;
 7         // while(l->left){//ERROR! CORRECT:while(l)
 8         //     ++llen;
 9         //     l = l->left;
10         // }
11         // while(r->right){
12         //     ++rlen;
13         //     r = r->right;
14         // }
15         for(; l!=NULL; ++llen, l=l->left); //ATTENTION!
16         for(; r!=NULL; ++rlen, r=r->right);
17         return llen == rlen ? (1<<llen) - 1 : 1 + countNodes(root->left) + countNodes(root->right);//ATTENTION
18         
19         
20     }
21 };

根据最后一行的节点尽可能靠左这一特点,先计算root->left->left->...这一路径长和root->right->right->...这一路径长。如果相等,则整棵树都是满的,利用等比数列公式(注意用了位运算);不相等则递归。

以上是关于LeetCode 222. Count Complete Tree Nodes的主要内容,如果未能解决你的问题,请参考以下文章

LeetCode222——Count Complete Tree Nodes

Leetcode 222. Count Complete Tree Nodes

LeetCode OJ 222. Count Complete Tree Nodes

[LeetCode] 222. Count Complete Tree Nodes Java

[Leetcode] Binary search -- 222. Count Complete Tree Nodes

leetcode 222.Count Complete Tree Nodes