题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
class Solution {
public:
vector<int> PrintFromTopToBottom(TreeNode* root) {
vector<int> ans;
vector<TreeNode*> res;
if(root == NULL)
return ans;
res.push_back(root);
for(int i=0;i<res.size();++i){
root=res[i];
ans.push_back(root->val);
if(res[i]->left)
res.push_back(res[i]->left);
if(res[i]->right)
res.push_back(res[i]->right);
}
return ans;
}
};
使用队列的版本:
/*
vector<TreeNode*> res;
上句里的res可以用队列实现,更舒服一点。
*/
class Solution {
public:
vector<int> PrintFromTopToBottom(TreeNode *rt) {
queue<TreeNode*> q;
q.push(rt);
vector<int> r;
while(!q.empty()){
rt = q.front(); q.pop();
if(!rt) continue;
r.push_back(rt -> val);
q.push(rt -> left);
q.push(rt -> right);
}
return r;
}
};