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
}