NO-22 从上往下打印二叉树
题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
输入:
{5,4,#,3,#,2,#,1}
输出
[5,4,3,2,1]
一刷
解法1:迭代做法,借助队列
vector<int> PrintFromTopToBottom(TreeNode* root) {
vector<int> result;
if (root == nullptr) return result;
queue<TreeNode*> q;
q.push(root);
TreeNode* node;
while (!q.empty()) {
node = q.front();
result.push_back(node->val);
if (node->left) q.push(node->left);
if (node->right) q.push(node->right);
q.pop();
}
return result;
}
二刷
class Solution {
public:
//本题是接用一个队列按层对数据进行入队出队操作
vector<int> PrintFromTopToBottom(TreeNode* root) {
if(root==nullptr)return vector<int>();
queue<TreeNode*>q;
q.push(root);
TreeNode *node=nullptr;
vector<int>result;
while(!q.empty()){
node=q.front();//引用队列的第一个元素 然后出队
q.pop(); // 删除
result.push_back(node->val);//push_back 想向量尾部增加一个元素 就是把上一步pop的结果保存到result中
if(node->left) q.push(node->left); //取根节点的左边,继续想队列中增加元素 push 增加元素
if(node->right) q.push(node->right);
}
return result;
}
};