算法
(BFS) $O(n)$
宽度优先遍历,一层一层来做。即:
- 将根节点插入队列中;
- 创建一个新队列,用来按顺序保存下一层的所有子节点;
- 对于当前队列中的所有节点,按顺序依次将儿子插入新队列;
- 按从左到右、从右到左的顺序交替保存队列中节点的值;
- 重复步骤2-4,直到队列为空为止。
时间复杂度
树中每个节点仅会进队出队一次,所以时间复杂度是 $O(n)$。
C++ 代码
/**
* 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<int> get_val(vector<TreeNode*> level)
{
vector<int> res;
for (auto &u : level)
res.push_back(u->val);
return res;
}
vector<vector<int>> printFromTopToBottom(TreeNode* root) {
vector<vector<int>>res;
if (!root) return res;
vector<TreeNode*>level;
level.push_back(root);
res.push_back(get_val(level));
bool zigzag = true;
while (true)
{
vector<TreeNode*> newLevel;
for (auto &u : level)
{
if (u->left) newLevel.push_back(u->left);
if (u->right) newLevel.push_back(u->right);
}
if (newLevel.size())
{
vector<int>temp = get_val(newLevel);
if (zigzag)
reverse(temp.begin(), temp.end());
res.push_back(temp);
level = newLevel;
}
else break;
zigzag = !zigzag;
}
return res;
}
};
赞,这样写更简洁一些
思路和“分行从上往下打印二叉树一样”,只是每次都判断一下是否要reverse
可以的,这个思路和上面的一样hh
比较疑惑,为什么高赞的解都没有使用reverse?
我看到了一种不用reverse的做法,可以用两个stack来交替做,作用和reverse是一样的~