题意描述:二叉树的层次遍历,奇数层从左往右遍历,偶数层从右往左遍历。
解法:广度优先搜索。
一般的层次遍历,用一个队列进行广度优先搜索即可。但是这个题目中的要求有所不同。在遍历过程中,需要纪录层数,根据层数采取相应的措施。因此定义了一个新的结构来储存结点,在原来结点的基础上加上一个字段表示层数:
struct Node {
int val;
int level;
TreeNode* left;
TreeNode* right;
Node(TreeNode* node, int l) : val(node->val),
left(node->left), right(node->right), level(l) {}
};
添加了层数信息之后就比较好解决了:在遍历过程中,判断当前从队列中取出的结点是否属于当前层次。若是,将结点值加入该层的结果集合;若不是,添加一个vector表示新的层,然后再加入该层。
遍历结束之后,所有层次的数据都是从左往右排列的,因此要将奇数层的vector翻转,使其为从右往左的序列。
最终代码为:
class Solution {
public:
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
if (root == NULL) return vector<vector<int> >();
Node _root(root, 0);
queue<Node> que;
que.push(_root);
vector<vector<int> > vec;
int count = 0; // 纪录vec中的vector<int>的数量
while (!que.empty()) {
Node t = que.front();
que.pop();
if (count <= t.level) {
vec.push_back(vector<int>());
count += 1;
}
vec[count - 1].push_back(t.val);
if (t.left != NULL) {
Node l(t.left, t.level + 1);
que.push(l);
}
if (t.right != NULL) {
Node r(t.right, t.level + 1);
que.push(r);
}
}
int size = vec.size();
for (int i = 1; i < size; i += 2) {
vec[i] = vector<int>(vec[i].rbegin(), vec[i].rend());
}
return vec;
}
};