题目链接
题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
解题思路
层序遍历 + 双端队列
利用双端队列的两端皆可添加元素的特性,设打印列表(双端队列) tmp
,并规定:
- 当前为奇数层 则添加至
tmp
尾部 , - 当前为偶数层 则添加至
tmp
头部 。
/**
* 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>> levelOrder(TreeNode* root) {
if(root==NULL) return {};
deque<TreeNode*> dq;
vector<vector<int>> ret;
// 记录当前队列中的单双层
// true为单层,false为双层
bool flag = true;
dq.push_back(root);
while(!dq.empty()){
vector<int> ans;
int sz = dq.size();
if(flag){
// 单层在双端队列的前面读取数据并将子结点在双端队列的后面插入队列
// 先左节点后右结点
while(sz--){
TreeNode* node = dq.front();
ans.push_back(node->val);
dq.pop_front();
if(node->left!=NULL) dq.push_back(node->left);
if(node->right!=NULL) dq.push_back(node->right);
}
}else{
// 双层在双端队列的后面读取数据并将子结点在双端队列的前面插入队列
// 先右节点后左结点
while(sz--){
TreeNode* node = dq.back();
ans.push_back(node->val);
dq.pop_back();
if(node->right!=NULL) dq.push_front(node->right);
if(node->left!=NULL) dq.push_front(node->left);
}
}
ret.push_back(ans);
// 记录单双层
flag = !flag;
}
return ret;
}
};
- 时间复杂度 O(N) : N 为二叉树的节点数量,即 BFS 需循环 N 次,占用 O(N) ;双端队列的队首和队尾的添加和删除操作的时间复杂度均为 O(1) 。
- 空间复杂度 O(N) : 最差情况下,即当树为满二叉树时,最多有 N/2 个树节点 同时 在 deque 中,使用 O(N) 大小的额外空间。