LeetCode - #199 二叉树的右视图

我报名参加金石计划1期挑战——瓜分10万奖池，这是我的第1篇文章，点击查看活动详情

前言

我们社区陆续会将顾毅（Netflix 增长黑客，《iOS 面试之道》作者，ACE 职业健身教练。）的 Swift 算法题题解整理为文字版以方便大家学习与阅读。

LeetCode 算法到目前我们已经更新到 198 期，我们会保持更新时间和进度（周一、周三、周五早上 9:00 发布），每期的内容不多，我们希望大家可以在上班路上阅读，长久积累会有很大提升。

不积跬步，无以至千里；不积小流，无以成江海，Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言，我们会尽力满足大家的需求。

难度水平：中等

1. 描述

给定一个二叉树的 根节点 root，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。

2. 示例

示例 1

输入: [1,2,3,null,5,null,4]
输出: [1,3,4]

示例 2

输入: [1,null,3]
输出: [1,3]

示例 3

输入: []
输出: []

提示：

• 二叉树的节点个数的范围是 [0,100]
• -100 <= Node.val <= 100

3. 答案

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     public var val: Int
 *     public var left: TreeNode?
 *     public var right: TreeNode?
 *     public init(_ val: Int) {
 *         self.val = val
 *         self.left = nil
 *         self.right = nil
 *     }
 * }
 */
 
class BinaryTreeRightSideView {
    func rightSideView(_ root: TreeNode?) -> [Int] {
        var res = [Int]()
        var nodeQ = [TreeNode]()
        
        if let root = root {
            nodeQ.append(root)
        }
        
        while nodeQ.count > 0 {
            var size = nodeQ.count
            
            for i in 0..<size {
                let node = nodeQ.removeFirst()
                
                if i + 1 == size {
                    res.append(node.val)
                }
                
                if let left = node.left {
                    nodeQ.append(left)
                }
                if let right = node.right {
                    nodeQ.append(right)
                }
            }
        }
        
        return res
    }
}

• 主要思想：使用一个队列来保持 TreeNode，并为每一层添加最后一个。
• 时间复杂度： O(n)
• 空间复杂度： O(n)

该算法题解的仓库：LeetCode-Swift

点击前往 LeetCode 练习

关于我们

我们是由 Swift 爱好者共同维护，我们会分享以 Swift 实战、SwiftUI、Swift 基础为核心的技术内容，也整理收集优秀的学习资料。

后续还会翻译大量资料到我们公众号，有感兴趣的朋友，可以加入我们。