题解 | 按之字形顺序打印二叉树:每层访问完根据条件reverse
按之字形顺序打印二叉树
https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0
每层访问完根据条件reverse
思路
根据层次遍历思想,设置队列,同时设置一个布尔变量判断这个数组是否进行元素反转
布尔变量可以实现两个值来回变化,每层存储后,都对其进行取非操作
知识点
反转函数reverse用法:
| 反转整个容器 | vector |
| 反转部分区间 [i, j) |
|
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
#include <algorithm>
#include <queue>
#include <vector>
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pRoot TreeNode类
* @return int整型vector<vector<>>
*/
vector<vector<int> > Print(TreeNode* pRoot) {
// write code here
vector<vector<int>> res;
if (pRoot == nullptr) return res;
queue<TreeNode*> q;
q.push(pRoot);
bool LeftToRight = true;
while (!q.empty()) {
vector<int> level;
int size = q.size();
for (int i = 0; i < size; i++) {
TreeNode* node = q.front();
q.pop();
level.push_back(node->val);
if (node->left) q.push(node->left);
if (node->right) q.push(node->right);
}
if (!LeftToRight) reverse(level.begin(), level.end());
res.push_back(level);
LeftToRight = !LeftToRight;
}
return res;
}
};
查看15道真题和解析