二叉树中和为某一值的路径 AcWing (JAVA)

输入一棵二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。

从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

保证树中结点值均不小于 0。

数据范围:

树中结点的数量 [0,1000]。

样例:

给出二叉树如下所示,并给出num=22。
      5
     / \
    4   6
   /   / \
  12  13  6
 /  \    / \
9    1  5   1

输出:[[5,4,12,1],[5,6,6,5]]

解题思路:核心方法思路依旧是 dfs 和 回溯。需要变通的是此次探索的新方向不再是左右上下,而是结点的左孩子和右孩子。定义两个 List 变量,一个用来储存符合条件的路径,一个用来储存正确答案。

值得注意的是:List 变量比较特别,采用,引用调用,如果不把正确路径copy一份再放入第二个List量中的话,后续回溯,最后只会看到第二个List储存的只有空(如果对 List 动态数组还不是很了解可以看:对 List 函数底层逻辑的理解

理论成立代码如下:

class Solution {
	 List<List<Integer>> s = new ArrayList<>();//储存答案
	 List<Integer> s1 = new ArrayList<>();//储存路径
	    public List<List<Integer>> findPath(TreeNode root, int sum) {
	    	dfs(root, 0, sum);  
	    	return s;
	    }
	    public void dfs(TreeNode root, int sum, int target) {
	    	if(root == null) return;//没有路径
	    	s1.add(root.val);
	    	sum = sum + root.val;
	    	if(root.left == null && root.right == null) {//叶子结点
	    		if(sum == target) {//符合条件
	    			List<Integer> temp = new ArrayList<>();
	    			temp.addAll(s1);//copy一份s1
	    			s.add(temp);//储存
	    		}
	    	}
	    	else {
	    		if(root.left != null) dfs(root.left, sum, target);//探索新结点
	    		if(root.right != null) dfs(root.right, sum, target);
	    	}
	    	s1.remove(s1.size() - 1);//回溯
	    }
	}
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值