剑指offer-22:从上往下打印二叉树
从根节点开始按宽度优先的顺序遍历整棵树,每次先扩展左儿子,再扩展右儿子;具体步骤如下:
1.先扩展根节点;2.再依次扩展根节点的左右儿子,也就是从左到右扩展第二层节点;3.再依次从左到右扩展第三层节点;4.依次类推
可以发现BFS的顺序就是题目要求的顺序,由于所有节点只被遍历一次,因此时间复杂度为O(n)。
具体代码如下:
class Solution {
public:
vector<int> PrintFromTopToBottom(TreeNode* root) {
vector<int> res;
if(!root) return res;
queue<TreeNode*> q;
q.push(root);
while(q.size()){
auto t = q.front(); //opps
q.pop();
res.push_back(t->val); //opps
if(t->left) q.push(t->left);
if(t->right) q.push(t->right);
}
return res;
}
};