【二叉树】二叉树的右视图

0x00 题目

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

---

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 rightSideView(_ root: TreeNode?) -> [Int] {
    guard let r = root else { return [] }
    
    // 保存结果 
    var res: [Int] = []
    
    // 存储每一层的节点
    var queue: [TreeNode] = []
    queue.append(r)

    while (!queue.isEmpty) {
        // 获取最后一个，保存
        let node = queue.last!
        res.append(node.val)

		// 获取下一层的所有节点
        var tmp: [TreeNode] = []
        for n in queue {
            if let left = n.left {
                tmp.append(left)
            }
            if let right = n.right {
                tmp.append(right)
            }
        }
        
        // 切换到下一层
        queue = tmp
    }
    
	// 返回结果  
    return res
}

---

小程序应用

---