从上往下打印出二叉树的每个节点,同层节点从左至右打印。
个人思考:使用一个队列一个栈一个存放节点,一个存放值。先将根节点加入到队列中,然后遍历队列中的元素,遍历过程中,访问该元素的左右节点,再将左右子节点加入到队列中来。
假设有如下二叉树:
输出结果应为 :1 2 3 4 5 6先检查是否为空树,是空树直接返回;
将根节点插入队列q里: 1
用tmp遍历该树,tmp=q.front();tmp初始指向根节点,
接下来:q.pop();
s.push_back(tmp->val) s此时:1
最后插入tmp左右节点:
if(tmp->left)
q.push(tmp->left);
if(tmp->right)
q.push(tmp->right);
此时q: 2 3
依次往下;
代码:
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> s;
queue<TreeNode*> q;//存放上层节点
if(root==NULL)//存放已访问的当前节点
return s;
q.push(root);
TreeNode* tmp;
while(!q.empty()){
tmp=q.front();
q.pop();
s.push_back(tmp->val);
if(tmp->left)
q.push(tmp->left);
if(tmp->right)
q.push(tmp->right);
}
return s;
}
};