【leetcode 二叉树】222. Count Complete Tree Nodes

笨比master

于 2021-01-24 19:15:59 发布

阅读量70

点赞数

分类专栏： leetcode 文章标签：二叉树完全二叉树节点计数广度优先搜索 C++

本文链接：https://blog.csdn.net/m0_37454852/article/details/113095294

版权

leetcode 专栏收录该内容

288 篇文章 0 订阅

订阅专栏

222. Count Complete Tree Nodes

在这里插入图片描述

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int countNodes(TreeNode* root) {
        queue<TreeNode*> Q;
        TreeNode* p;
        if(root) Q.push(root);
        int cnt = 0;
        while(!Q.empty()) {
            p = Q.front();
            Q.pop();
            if(p->left) Q.push(p->left);
            if(p->right) Q.push(p->right);
            cnt++;
        } 
        return cnt;
    }
};