leetcode 222——完全二叉树节点个数

原创于 2019-07-14 16:20:02 发布 · 98 阅读

0 ·

CC 4.0 BY-SA版权

C++ 同时被 2 个专栏收录

38 篇文章

订阅专栏

数据结构与算法

17 篇文章

订阅专栏

博客提及了二分法求解，同时指出最暴力的递归方法也能通过，主要围绕这两种求解方法展开。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

二分法求解。

/**
 * 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:
    int countNodes(TreeNode* root) {
        if(root == nullptr) return 0;
        
        TreeNode *t = root, *prev_tree = root;
        int total = 0;
        int h = -1;
        while(t != nullptr) {
            t = t->left;
            h++;
        }
        total = (0x1 << h) - 1;
        t = root;
        int ch = 0;
        int leftnode = 0;
        while(ch < h){
            prev_tree = t;
            t = t->right; ch++;
            
            for(int i = 0; i < h - ch; i++) t = t->left; 
            
            if(t == nullptr){
                t = prev_tree->left;
            } else {
                t = prev_tree->right;
                leftnode = leftnode +  (0x1 << h - ch);
            }
        }
        
        return total + leftnode + ((t == nullptr) ? 0 : 1);
    }
};

当然，最暴力的递归方法也可以通过：

int find(TreeNode *root){
    if(!t) return 0;
    return 1 + find(t->left) + find(t->right);
}