从上到下按层打印二叉树,同一层的结点按从左到右的顺序打印,每一层打印到一行。
样例
输入如下图所示二叉树[8, 12, 2, null, null, 6, null, 4, null, null, null]
8
/ \
12 2
/
6
/
4
输出:[[8], [12, 2], [6], [4]]
分析:
每一层结束的时候,往queue里塞一个NULL做标记。
在queue里读取一个数出来之后,先看看是不是行标识符NULL
(因为是BFS,当前行读完,下一行有哪些要读的也都放在queue里了,可以在queue结尾给加一个新的NULL), 是的话再看看是不是整个树读完了(即queue里没有点了)。
时间复杂度分析:每个点遍历一次,所以时间复杂度是 O(n)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> printFromTopToBottom(TreeNode* root) {
vector<vector<int>> res;
queue<TreeNode*> q;
if (!root) return res;
q.push(root);
q.push(nullptr);//存放NULL作为当前行的行末标记
vector<int> cur; // 用于存放当前行的元素
while(q.size()) {
auto x = q.front();
q.pop();
if (x) {
cur.push_back(x->val);
if(x->left) q.push(x->left);
if(x->right) q.push(x->right);
} else {//访问到行末标记null
if(q.size()) q.push(nullptr); //增加行末标记
res.push_back(cur); //当前行存放结果数组中
cur.clear(); //把当前行清空
}
}
return res;
}
};
网友评论