面试题34:二叉树中和为某一值的路径

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

思路:先序遍历树,用全局变量LinkedList类型的 path 来记录路径,暴力回溯,tar 不停的递减,直到 tar 为 0 并且当前结点为叶子结点,那么找到正确的路径 path,将其存入结果集中。

  • 时间复杂度O(n),遍历书中每个结点的时间复杂度
  • 空间复杂度O(n),需要定义一个辅助数组 path ,来存储路径,结果集不算再空间复杂度里面。
class Solution {
    // 定义两个全局变量比较好存储并且返回,不用通过传参或者函数返回值来修改
    public LinkedList<List<Integer>> res = new LinkedList<>();
    public LinkedList<Integer> path = new LinkedList<>();
    
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        recur(root, sum);
        return res;
    }
    
    // 不要写成 static ,不然里面定义的成员都是 static 的,全局变量的多个测试用例就会发生错误
    void recur(TreeNode root, int tar) {
        if (root == null) {
            return ;
        }
        path.add(root.val);
        tar -= root.val;
        if (tar == 0 && root.left == null && root.right == null) {
            res.add(new LinkedList(path));
        }
        recur(root.left, tar);
        recur(root.right, tar);
        // 因为 path 是全局变量,当前结点遍历完了,一定要进行回溯,
        path.removeLast();
        return ;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值