请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[20,9],
[15,7]
]
提示:
节点总数 <= 1000
解题思路
奇偶层的打印顺序不一样是相反的,可以利用层数偶数与否调用reverse来解决,但是海量数据时这个效率很低,不推荐。
因为奇数层的打印顺序是从左到右,偶数层的打印顺序是从右到左,可以利用STL容器deque中
push_back(),push_front(),front(),back(),pop(),popfront()来实现
前取后放,后取前放
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> res;
if (root==NULL)//根节点为空时
return res;
bool flag = true; //从左向右打印为true,从右向左打印为false
deque<TreeNode*> q;//存取元素的队列
q.push_back(root);//根节点入队列
while (!q.empty())//队列非空时执行
{
int n = q.size();//队列长度
vector<int> out;//暂存一维数组
TreeNode* node;//二叉树
while (n>0)
{
//奇数层要求从左向右输出,偶数层反之
//step 1.一开始flag==ture时,从前边取奇数层(即是根节点),
//然后存下一层(偶数层)的节点顺序地放在队列尾部(此层节点顺序排列);
//step 2.到偶数层从尾部取出(存放在数组中就是逆序了),
//然后存下一层(奇数层)的节点逆序地放在队列头部(此层节点顺序排列);
//因此,按照此逻辑一层层遍历可以得到结果
if (flag) // 前取后放:从左向右打印,所以从前边取,后边放入
// 从队头取出要打印的,左右子节点正序从队尾加入
{
node = q.front();
q.pop_front();
if (node->left)
q.push_back(node->left); // 下一层顺序存放至尾
if (node->right)
q.push_back(node->right);
}
else //后取前放: 从右向左,从后边取,前边放入
// 从队尾取出要打印的,左右子节点逆序从队头加入
{
node = q.back();
q.pop_back();
if (node->right)
q.push_front(node->right); // 下一层逆序存放至首
if (node->left)
q.push_front(node->left);
//存了奇数层的节点在前面,按照从左向右的顺序方式
}
out.push_back(node->val);
n--;//n自减
}
flag = !flag; // 改变打印顺序
res.push_back(out);// 打印完了一层
}
return res;
}
};
复杂度分析:
时间复杂度 O(N):
空间复杂度 O(N) :
作者:bryceustc
链接:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof/solution/cli-yong-dequeshuang-duan-dui-lie-hao-shi-0ms-ji-b/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。