LeetCode 0103.二叉树的锯齿形层序遍历：层序遍历 + 适时翻转

这篇具有很好参考价值的文章主要介绍了LeetCode 0103.二叉树的锯齿形层序遍历：层序遍历 + 适时翻转。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

【LetMeFly】103.二叉树的锯齿形层序遍历：层序遍历 + 适时翻转

力扣题目链接：https://leetcode.cn/problems/binary-tree-zigzag-level-order-traversal/

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

示例 1：

LeetCode 0103.二叉树的锯齿形层序遍历：层序遍历 + 适时翻转,题解,# 力扣LeetCode,leetcode,算法,题解,二叉树,层序遍历

输入：root = [3,9,20,null,null,15,7]
输出：[[3],[20,9],[15,7]]

示例 2：

输入：root = [1]
输出：[[1]]

示例 3：

输入：root = []
输出：[]

提示：

树中节点数目在范围 [0, 2000] 内
-100 <= Node.val <= 100

方法一：层序遍历 + 适时翻转

前言

相信大家已经做过关于层序遍历的练习了，没有做过的建议先做一做，可以参考这篇题解：LeetCode 102.二叉树的层序遍历 + 针对C++的使用空间优化（可能不同于常规做法）

思路

这道题与之不同的是，每隔一层需要变化一下遍历方向。

真的要变换遍历方向吗？其实不然：

我们只需要在正常遍历的基础上使用一个变量来记录当前这一层是否需要翻转，若需翻转则在这一层遍历结束后reverse一下最后一层即可。

时间复杂度 $O (N)$ ，其中 $N$ 是节点个数
空间复杂度 $O (N 2)$ ，其中 $N 2$ 是节点最多的一层的节点数

AC代码

C++

class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        vector<vector<int>> ans;
        bool ifReverse = false;
        queue<TreeNode*> q;
        if (root) {
            q.push(root);
        }
        while (q.size()) {
            ans.push_back({});
            for (int _ = q.size(); _ > 0; _--) {
                TreeNode* thisNode = q.front();
                q.pop();
                ans.back().push_back(thisNode->val);
                if (thisNode->left) {
                    q.push(thisNode->left);
                }
                if (thisNode->right) {
                    q.push(thisNode->right);
                }
            }
            if (ifReverse) {
                reverse(ans.back().begin(), ans.back().end());
            }
            ifReverse = !ifReverse;
        }
        return ans;
    }
};

Python

# from typing import List, Optional

# # Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right

class Solution:
    def zigzagLevelOrder(self, root: Optional[TreeNode]) -> List[List[int]]:
        ans = []
        ifReverse = False
        q = []
        if root:
            q.append(root)
        while q:
            ans.append([])
            for _ in range(len(q)):
                thisNode = q[0]
                q = q[1:]
                ans[-1].append(thisNode.val)
                if thisNode.left:
                    q.append(thisNode.left)
                if thisNode.right:
                    q.append(thisNode.right)
            if ifReverse:
                ans[-1].reverse()
            ifReverse = not ifReverse
        return ans