关闭

Count Complete Tree Nodes

117人阅读 评论(0) 收藏 举报
分类:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void traverseLeft(TreeNode* root, int& h)
    {
        if(root == NULL)
        {
            return;
        }
        ++ h;
        traverseLeft(root -> left, h); 
        
    }
    void traverseRight(TreeNode* root, int& h)
    {
        if(root == NULL)
        {
            return;
        }
        ++ h;
        traverseRight(root -> right, h);
    }

    int countNodes(TreeNode* root) 
    {
        if(root == NULL)
        {
            return 0;
        }
        int h1 = 1;
        int h2 = 1;
        traverseLeft(root -> left, h1);
        traverseRight(root -> right, h2);
        if(h1 == h2)
        {
            return pow(2, h1) - 1;
        }
        //int h = max(h1, h2);

        return countNodes(root -> left) + countNodes(root -> right) + 1;

    }
};


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:81455次
    • 积分:3659
    • 等级:
    • 排名:第8881名
    • 原创:295篇
    • 转载:27篇
    • 译文:0篇
    • 评论:3条
    最新评论