[leetcode]222. Count Complete Tree Nodes

题目地址

https://leetcode.com/problems/count-complete-tree-nodes/

题目大意

求一个完全二叉树的节点数量

解题思路

最简单的是遍历数节点，复杂度O(n)，不出意外的TLE了。考虑到时完全二叉树，所以可以通过比较最左叶子节点，和最右叶子节点的高度，来判断是不是满二叉树，如果是满二叉树，则直接能求出节点个数。如果不是则分别对左右子树做此递归操作。时间复杂度是O(h^2)，可以AC。

代码

class Solution {
public:
    int countNodes(TreeNode* root) {
        if (root == NULL) {
            return 0;
        }

        int l = leftHigh(root);
        int r = rightHigh(root);

        if (l == r) {
            return (2 << (l - 1)) - 1;
        } else {
            return countNodes(root->left) + countNodes(root->right) + 1;
        }
    }

private:
    int leftHigh(TreeNode * root) {
        int h = 0;
        while (root != NULL) {
            root = root->left;
            ++h;
        }
        return h;
    }

    int rightHigh(TreeNode * root) {
        int h = 0;
        while (root != NULL) {
            root = root->right;
            ++h;
        }
        return h;
    }
};