【二叉树】二叉树的锯齿形层序遍历

0x00 题目

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


0x01 思路

使用 层序遍历 的方式即可
可以使用一个布尔变量记录方向
来控制 从左往右,还是 从右往左


0x02 解法

语言:Swift

树节点:TreeNode

public class TreeNode {
    public var val: Int
    public var left: TreeNode?
    public var right: TreeNode?
    public init() { self.val = 0; self.left = nil; self.right = nil; }
    public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

解法:

func zigzagLevelOrder(_ root: TreeNode?) -> [[Int]] {
    guard let root = root else { return [] }

    // 存放结果 
    var res: [[Int]] = []
    
    // 存放每一层的节点
    var queue: [TreeNode] = [root]
    
    // 转换标识
    var flag: Bool = false 

    while !queue.isEmpty {
        // 存放下一层的节点
        var tmp: [TreeNode] = []
        // 存放当前层的节点值
        var arr: [Int] = []

        while !queue.isEmpty {
            let node = queue.removeFirst()
            arr.append(node.val)
            
            // 左节点存在
            if let left = node.left {
                tmp.append(left)
            }
            
            // 右节点存在            
            if let right = node.right {
                tmp.append(right)
            }
        }

        // 需要转换
        if flag {
            arr.reverse()
        }
        
        // 切换到下一层
        queue = tmp
        res.append(arr)
        
        // 标志切换
        flag = !flag
    }
    
    return res
}


祝大伙除夕快乐,身体健康,万事如意!


上一篇:二叉树基本操作和相关列题


下一篇:flink写入es