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 binaryTreePaths(_ root: TreeNode?) -> [String] {
guard let root = root else { return [] }
// 存放所有路径的数组
var res: [String] = []
// 存放当前路径的数组
var tmp: [String] = []
// 递归函数
func build(_ root: TreeNode?) {
// 没路了,返回
if root == nil {
return
}
// 添加节点
tmp.append("\(root!.val)")
// 往下走
build(root?.left)
build(root?.right)
// 叶子节点,走到底了
if root?.left == nil && root?.right == nil {
// 存放结果
res.append(tmp.joined(separator:"->"))
}
// 移除最后一步,回头找另外的路线
tmp.removeLast()
}
// 开始下山啦~
build(root)
return res
}
解法二:
递归的同时生成路径
func binaryTreePaths(_ root: TreeNode?) -> [String] {
guard let root = root else { return [] }
var res: [String] = []
func build(_ root: TreeNode?, _ path: String) {
// 叶子节点,走到底了
if root?.left == nil && root?.right == nil {
// 存放结果
res.append(path)
return
}
// 左边有路,往左边走
if let left = root?.left {
build(left, path + "->\(left.val)")
}
// 右边有路,往右边走
if let right = root?.right {
build(right, path + "->\(right.val)")
}
}
// 开始下山啦~
build(root, "\(root.val)")
return res
}
小笔记
笔记从未如此简单