给定一颗二叉树,依次从根节点一层一层返回节点
基本思路:利用队列模拟这一过程
根节点入队列
取队列头节点
若该节点还有左右节点 入队列
之道队列为空
二叉树结构
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
主过程 采用vector模拟队列
class Solution {
public:
vector<int> PrintFromTopToBottom(TreeNode* root) {
vector<int> res;
if(root==NULL)
return res;
vector<TreeNode*> queue;//队列
queue.push_back(root);
while(queue.empty()!=1)
{
TreeNode* now=queue[0];//取队列头部节点
queue.erase(queue.begin());
res.push_back(now->val);//放入结果中
if(now->left!=NULL)
queue.push_back(now->left);
if(now->right!=NULL)
queue.push_back(now->right);
}
return res;
}
};
进阶问题 要求单独每层的结果
思路:记录下一层的节点数目
vector<vector<int>> PrintFromTopToBottom(TreeNode* root) {
vector<vector<int>> res;
if(root==NULL)
return res;
vector<TreeNode*> queue;//队列
queue.push_back(root);
while(queue.empty()!=1)
{
int size=queue.size();//该层的长度
for(int i=0;i<size;i++) //取该层个数的节点
{
vector<int> one;
TreeNode* now=queue[0];//取队列头部节点
queue.erase(queue.begin());
one.push_back(now->val);//放入结果中
if(now->left!=NULL)
queue.push_back(now->left);
if(now->right!=NULL)
queue.push_back(now->right);
res.push_back(one);
}
}
return res;
}