113 路经总和(两种解法)

给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。

叶子节点 是指没有子节点的节点。

方法1:求解所有路径再求和

因为之前做过《257 二叉树的所有路径》,所以可以先递归求出所有路径,再判断哪一条路径的和为targetSum即可。该方法效率较低。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<List<Integer>> pathSum(TreeNode root, int targetSum) {
        List<List<Integer>> tmp = new ArrayList();
        List<Integer> paths = new ArrayList();
        List<List<Integer>> res = new ArrayList();
        getPaths(root,paths,tmp);
        for(int i=0;i<tmp.size();i++){
            if(tmp.get(i).stream().reduce(0,Integer::sum)==targetSum){
                res.add(tmp.get(i));
            }
        }
        return res;
    }
    public void getPaths(TreeNode root,List<Integer> paths,List<List<Integer>> res){
        if(root!=null){
            List<Integer> pathTmp = new ArrayList(paths);
            pathTmp.add(root.val);
            if(root.left==null&&root.right==null){
                res.add(pathTmp);
            }else{
                getPaths(root.left,pathTmp,res);
                getPaths(root.right,pathTmp,res);
            }
        }
    }
}

方法二:回溯法

dfs遍历树,记录路径上的和,与targetSum进行比较,关键在于,回溯到上一个节点的时候,需要从路径path中剔除该节点。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    List<List<Integer>> res = new ArrayList();
    LinkedList<Integer> path = new LinkedList();
    public List<List<Integer>> pathSum(TreeNode root, int targetSum) {
        recur(root,targetSum);
        return res;
    }
    public void recur(TreeNode root,int targetSum){
        if(root==null){
            return;
        }
        path.add(root.val);
        targetSum = targetSum-root.val;
        if(targetSum==0&&root.left==null&&root.right==null){
            res.add(new LinkedList<Integer>(path));
        }
        recur(root.left,targetSum);
        recur(root.right,targetSum);
        path.removeLast();
    }
}
  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值