【LeetCode 222_完全二叉树_遍历】Count Complete Tree Nodes

解法一：递归

int countNodes(TreeNode* root)
{
    if (root == NULL)
        return ;

    TreeNode *pLeft = root->left;
    TreeNode *pRight = root->right;
    int ldepth = , rdepth = ;
    while (pLeft) {
        ldepth++;
        pLeft = pLeft->left;
    }
    while (pRight) {
        rdepth++;
        pRight = pRight->right;
    }
    if (ldepth == rdepth)
        return ( << (ldepth + )) - ;
    else
        return countNodes(root->left) + countNodes(root->right) + ;
}

解法二：迭代

int GetDepth(TreeNode *root)
{
    int depth = ;
    while (root) {
        depth++;
        root = root->left;
    }
    return depth;
}

int countNodes(TreeNode* root)
{
    if (root == NULL)
        return ;

    int depth = GetDepth(root);
    int leaf = ;
    int depth_left, depth_right;
    while (true) {
        depth_left = GetDepth(root->left);
        depth_right = GetDepth(root->right);
        if (depth_left ==  && depth_right == ) {
            leaf += ;
            break;
        }

        if (depth_left == depth_right) {
            leaf +=  << (depth_left - );
            root = root->right;
        } else {
            root = root->left;
        }
    }
    return ( << (depth - )) -  + leaf;
}