题目
给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回锯齿形层次遍历如下:
[
[3],
[20,9],
[15,7]
]
思路
进行层次遍历,用二维数组保存数据,其中每一层获得的数据放在同一个数组。然后遍历二维数组,将序号为奇数的数组倒过来。
C++代码
class Solution {
public:
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
vector<vector<int>> items;
travel(root, 0, items);
for (int i = 0; i < items.size();++i) {
if (i % 2 == 1) {
reverse(items[i].begin(), items[i].end());
}
}
return items;
}
void travel(TreeNode * node, int depth,vector<vector<int>> & items) {
if (node == NULL) return;
if (items.size() == depth) items.push_back({});
items[depth].push_back(node->val);
travel(node->left, depth+1, items);
travel(node->right, depth+1, items);
}
};
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal