思路:本题其实挺简单的,只需要用一个二叉树的层序遍历,挨个记录节点个数就行了。
class Solution {
public:
int countNodes(TreeNode* root) {
int cnt=0;
vector<vector<int>> ret;
queue<TreeNode*>que;
if(root)
{
que.push(root);
}
while(!que.empty())
{
vector<int> vec;
int size=que.size();
while(size--)
{
TreeNode* node=que.front();
vec.push_back(node->val);
cnt++;
que.pop();
if(node->left) que.push(node->left);
if(node->right) que.push(node->right);
}
ret.push_back(vec);
}
return cnt;
}
};