LeetCode 103. 二叉树的锯齿形层次遍历

LeetCode 103. 二叉树的锯齿形层次遍历

题目

给定一个二叉树，返回其节点值的锯齿形层次遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。

例如：
给定二叉树 [3,9,20,null,null,15,7],

3

/
9 20
/
15 7
返回锯齿形层次遍历如下：

[
[3],
[20,9],
[15,7]
]

分析

还是二叉树，还是遍历，不过这次似乎只能用bfs。将每次的结果放入一个容器中存储起来，这样能保证下一层遍历的有序。然后整体跟上一题差不多。

C++代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
                

class Solution {
public:
    vector<vector<int>> res;
    vector<vector<TreeNode*>> st;
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        helper(root, 0);
        for(int i = 0; i < res.size(); i++){
            if(i % 2 == 1){
                for(int j = res[i].size() - 1; j >= 0; j--){
                    helper(st[i][j]->left, i + 1);
                    helper(st[i][j]->right, i + 1);
                }
            }else{
                for(int j = res[i].size() - 1; j >= 0; j--){
                    helper(st[i][j]->right, i + 1);
                    helper(st[i][j]->left, i + 1);
                }
            }
        }
        return res;
    }
    void helper(TreeNode* root, int level){
        if(root == NULL){
            return;
        }
        if(level >= res.size()){
            vector<int> temp;
            vector<TreeNode*> t;
            temp.push_back(root->val);
            t.push_back(root);
            res.push_back(temp);
            st.push_back(t);
        }else{
            res[level].push_back(root->val);
            st[level].push_back(root);
        }
    }
};