JAVA程序设计:路径总和 II(LeetCode:113)

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

说明: 叶子节点是指没有子节点的节点。

示例:
给定如下二叉树,以及目标和 sum = 22,

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1
返回:

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

思路:dfs搞搞就好了。

class TreeNode {
     int val;
     TreeNode left;
     TreeNode right;
	 TreeNode(int x) { val = x; }
}

class Solution {
    public List> pathSum(TreeNode root, int sum) {
        
    	List res=new ArrayList();
    	List> ans=new ArrayList>();
    	
    	if(root==null)
    		return ans;
    	
    	dfs(ans,res,root,sum,0);
    	
    	return ans;
    }
    
    private void dfs(List> ans,List res,TreeNode root,int targetsum,int sum)
    {
    	res.add(root.val);
    	if(root.left==null && root.right==null)
    	{
    		if(targetsum==sum+root.val)
    			ans.add(new ArrayList(res));
    	}
    	if(root.left!=null)
    		dfs(ans,res,root.left,targetsum,sum+root.val);
    	if(root.right!=null)
    		dfs(ans,res,root.right,targetsum,sum+root.val);
    	res.remove(res.size()-1);
    }
}

 

你可能感兴趣的:(JAVA程序设计:路径总和 II(LeetCode:113))