LeetCode习题:二叉树中和为某一值的路径

题目描述:输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

示例:
给定如下二叉树,以及目标和 target = 22
		  5
         / \
        4   8
       /   / \
      11  13  4
     /  \    / \
    7    2  5   1
返回:
	[
	  [5,4,11,2],
	  [5,8,4,5]
	]
提示: 节点总数 <= 10000
题解:先序遍历
解题思路:按照 “根、左、右” 的顺序,遍历树的所有节点。在遍历过程中,记录从根节点到当前节点的路径,在该路径中,各节点的值的和等于目标值时,将此路径加入结果列表中。
解题语言: Swift
/**
 * Definition for a binary tree node.
 * 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
 *     }
 * }
 */
class Solution {
    var resultPaths = [[Int]]()
    var currentPath = [Int]()

    func pathSum(_ root: TreeNode?, _ target: Int) -> [[Int]] {
        subPathSum(root, target)
        return resultPaths
    }

    func subPathSum(_ root: TreeNode?, _ target: Int) {
        guard let root = root else {
            return
        }
        currentPath.append(root.val)
        var newTarget = target - root.val
        if (newTarget == 0 && root.left == nil && root.right == nil) {
            resultPaths.append(currentPath)
        }
        subPathSum(root.left, newTarget)
        subPathSum(root.right, newTarget)
        currentPath.removeLast()
    }
}

复杂度分析:
时间复杂度:O(N),其中 N 是二叉树的节点数。 先序遍历需要遍历所有节点。
空间复杂度:O(N),最差情况下,即树退化为链表时,path 存储所有树节点,使用 O(N) 额外空间。
题目来源:LeetCode
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值