题解 | #按之字形顺序打印二叉树#
按之字形顺序打印二叉树
https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0
在层序遍历的基础上,加上zigzag来判断如果是false就从左到右,如果是true就从右到左
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pRoot TreeNode类
* @return int整型vector<vector<>>
*/
vector<vector<int> > Print(TreeNode* root) {
// write code here
vector<vector<int>> res;
queue<TreeNode*> q;
if (root) q.push(root);
bool zigzag = false; // 默认从左开始再到右
while(q.size()) {
vector<int> level; // 一层
int len = q.size();
// 层序遍历
while(len -- ) {
auto t = q.front(); // 取队首元素
q.pop(); // 删除队首元素
level.push_back(t->val); // 保存当前层的值
if (t->left) q.push(t->left); // 如果左边还有就存入队列
if (t->right) q.push(t->right); // 如果右边还有就存入队列
}
if (zigzag) reverse(level.begin(), level.end()); // 如果为false就是从左到右,如果是true就从右到左
res.push_back(level); // 答案存下 一层
zigzag = !zigzag; // 取反
}
return res;
}
};
#按之字形顺序打印二叉树#