题目

---

给定一个二叉树，返回该二叉树的之字形层序遍历，（第一层从左向右，下一层从右向左，一直这样交替）。

数据范围：0≤n≤1500,树上每个节点的val满足∣val∣<=1500

要求：空间复杂度：O(n)，时间复杂度：O(n)

例如：
给定的二叉树是{1,2,3,#,#,4,5}

该二叉树之字形层序遍历的结果是
[
[1],
[3,2],
[4,5]
]

示例1

输入：
{1,2,3,#,#,4,5}
返回值：
[[1],[3,2],[4,5]]
说明：
如题面解释，第一层是根节点，从左到右打印结果，第二层从右到左，第三层从左到右。

示例2

输入：
{8,6,10,5,7,9,11}
返回值：
[[8],[10,6],[5,7,9,11]]

示例3

输入：
{1,2,3,4,5}
返回值：
[[1],[3,2],[4,5]]

思路

---

用两个辅助栈，通过层序遍历二叉树，需要注意的是因为奇数层要从左往右打印，所以前一个偶数层入栈需要先右后左，偶数层正好相反。

总结规律就是奇数层先左后右入栈，偶数层先右后左入栈。

解答代码

---

/*** struct TreeNode {*	int val;*	struct TreeNode *left;*	struct TreeNode *right;*	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}* };*/
#include <stack>
#include <vector>
class Solution {
public:/*** @param pRoot TreeNode类 * @return int整型vector<vector<>>*/vector<vector<int> > Print(TreeNode* pRoot) {// write code herevector<vector<int> > res{};if (pRoot == nullptr) {return res;}stack<TreeNode*> odd_level;// 奇数层，从左往右打印stack<TreeNode*> even_level;// 偶数层，从右往左打印int cur = 1;// 当前层索引odd_level.push(pRoot);vector<int> tmp;while (!odd_level.empty() || !even_level.empty()) {if ((cur&1) == 1) {// 奇数层auto size = odd_level.size();for (int i = 0; i < size; i++) {auto node = odd_level.top();tmp.push_back(node->val);// 先左后右if (node->left != nullptr) {even_level.push(node->left);}if (node->right != nullptr) {even_level.push(node->right);}odd_level.pop();}} else {// 偶数层auto size = even_level.size();for (int i = 0; i < size; i++) {auto node = even_level.top();tmp.push_back(node->val);// 先右后左if (node->right != nullptr) {odd_level.push(node->right);}if (node->left != nullptr) {odd_level.push(node->left);}even_level.pop();}}++cur;res.push_back(tmp);tmp.clear();}return res;}
};