【力扣刷题练习】103. 二叉树的锯齿形层序遍历

NaturalHarmonia2024-01-21 23:58

题目描述：

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

题目解答：

cpp 复制代码

class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        queue<TreeNode*> q;
        if (root != nullptr)
            q.push(root);
        bool flag = false;
        vector<vector<int>> ans;
        while (!q.empty()) {
            int size = q.size();
            vector<int> v;
            for (int i = 0; i < size; i++) {
                TreeNode* t = q.front();
                q.pop();
                v.push_back(t->val);
                if (t->left)
                    q.push(t->left);
                if (t->right)
                    q.push(t->right);
            }
            if(flag)
                reverse(v.begin(),v.end());
            ans.emplace_back(v);
            flag = !flag;
        }
        return ans;
    }
};

题目思路：

本题思路由【力扣刷题练习】102. 二叉树的层序遍历演变而来

在原有思路以及代码下增加一个布尔型flag变量用以控制遍历顺序，每隔一层调用一次reverse翻转顺序最终实现锯齿形层次遍历。