从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路:
借助一个队列(在队尾添加元素,在队首取出元素的特性),向队列中的压入顺序就是层序遍历的顺序,然后一个个放在向量result中。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
vector<int> PrintFromTopToBottom(TreeNode* root) {
vector<int> result;//定义一个数组,方便返回
if(root==NULL){
return result;
}
queue<TreeNode*> Q;//定义一个队列
Q.push(root);//将根节点放入队列中
while(!Q.empty())
{
TreeNode* p = Q.front();//新的树节点p为队首元素
Q.pop();//删除
result.push_back(p->val);//将第一个元素存入数组中
if(p->left!=NULL)//当树的左孩子不是空,就压入队列中
Q.push(p->left);
if(p->right!=NULL)//当树的右孩子不为空,压入队列中
Q.push(p->right);
}
return result;
}
};