算法题之求所有从根节点到叶子节点路径总和等于给定目标和的路径问题

算法题之求所有从根节点到叶子节点路径总和等于给定目标和的路径问题

题目描述

在这里插入图片描述

思路图解

在这里插入图片描述
在这里插入图片描述

代码实现
方式1
	public List<List<Integer>> pathSum(TreeNode root, int sum) {
        //定义存储最后结果的list集合
        List<List<Integer>> res = new ArrayList<>();
        //定义sub存储每一次满足要求的结果集
        List<Integer> sub = new ArrayList<>();
        findPath(root,sum,res,sub);
        //此时res中保存所有满足条件的结果集
        return res;

    }
	private void findPath(TreeNode root, int sum, List<List<Integer>> res, List<Integer> sub) {
        if (root == null) {
            return;
        }
        //将当前结点的值加入sub
        sub.add(new Integer(root.val));
        //判断当前是否到达叶子结点
        if (root.left == null && root.right == null) {
            if (sum == root.val) {
                res.add(new ArrayList<>(sub));
            }
            //否则移除最后一个元素
            sub.remove(sub.size() - 1);
            return;
        }
        //如果没有到达叶子结点
        //递归左子树:此时将sum减去root.val
        findPath(root.left,sum - root.val,res,sub);
        //递归右子树:此时同样将sum减去root.val
        findPath(root.right,sum - root.val,res,sub);
        //回溯时去掉sub的最后一个值
        sub.remove(sub.size() - 1);
    }
方式2
	public List<List<Integer>> pathSum(TreeNode root, int sum) {
        //定义存储最后结果的list集合
        List<List<Integer>> res = new ArrayList<>();
        //定义sub存储每一次满足要求的结果集
        List<Integer> sub = new ArrayList<>();
        findPath(root,sum,res,sub);
        //此时res中保存所有满足条件的结果集
        return res;

    }
	//方法1:保持sum的值不变:在遍历到达二叉树的叶子结点时,比较sum值和sub中所有的val值得和是否相等
    //若相等,则将其加入res结果集中
    //若不相等:则去掉最后添加的元素,返回,进行回溯
    private void findPath(TreeNode root, int sum, List<List<Integer>> res, List<Integer> sub) {
        //当root结点为null直接返回
        if (root == null) {
            return;
        }

        //将当前结点的val加入到sub中
        sub.add(root.val);

        if (root.left == null && root.right == null) {//即到达树的叶子结点
            if (getSum(sub) == sum) {
                res.add(new ArrayList<>(sub));
            }
            return;
        }

        //递归查询该结点的左子树
        if (root.left != null) {
            findPath(root.left,sum,res,sub);
            sub.remove(sub.size() - 1);
        }

        //递归查询该结点的右子树
        if (root.right != null) {
            findPath(root.right,sum,res,sub);
            sub.remove(sub.size() - 1);
        }

    }

    private int getSum(List<Integer> sub) {
        int sum = 0;
        for (int i = 0; i < sub.size(); i++) {
            sum = sum + sub.get(i);
        }
        return sum;
    }
TreeNode节点类
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;


    public TreeNode(int val) {
        this.val = val;
    }
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值