[LeetCode]Path Sum II

最新推荐文章于 2017-06-29 10:05:00 发布

原创最新推荐文章于 2017-06-29 10:05:00 发布 · 566 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Java #LeetCode #Depth-first search #Tree

LeetCode 专栏收录该内容

160 篇文章

订阅专栏

本文介绍了一种算法，用于查找二叉树中所有从根节点到叶子节点的路径，其中路径上节点值的总和等于指定的目标值。通过递归方法实现，示例中目标和为22，返回所有符合条件的路径。

Given a binary tree and a sum, find all root-to-leaf paths where each path's sum equals the given sum.

For example:
Given the below binary tree and

sum
 = 22

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

return

[
   [5,4,11,2],
   [5,8,4,5]
]

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
	List<List<Integer>> res = new ArrayList<>();
	public List<List<Integer>> pathSum(TreeNode root, int sum){
        if(root==null) return res;
        pathSum(root,sum,0,new ArrayList<Integer>());
        return res;
    }
    
    private void pathSum(TreeNode root,int sum,int val,List<Integer> list){ 
    	List<Integer> la = new ArrayList<>(list);
    	la.add(root.val);
    	if((root.left==null&&root.right==null)){
    		if(val+root.val==sum) res.add(la);
    		return;
    	}
    	if(root.left!=null){
    		pathSum(root.left,sum,val+root.val,la);
    	}
    	if(root.right!=null){
    		pathSum(root.right,sum,val+root.val,la);
    	}
    }
}