二叉树的路径和

来源:互联网 发布:淘宝仿货做的很好的店 编辑:程序博客网 时间:2024/06/05 23:39

给定一个二叉树,找出所有路径中各节点相加总和等于给定 目标值 的路径。

一个有效的路径,指的是从根节点到叶节点的路径。

样例

给定一个二叉树,和 目标值 = 5:

     1    / \   2   4  / \ 2   3

返回:

[  [1, 2, 2],  [1, 4]]
代码如下:

/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */
public class Solution {
    /**
     * @param root the root of binary tree
     * @param target an integer
     * @return all valid paths
     */
    public List<List<Integer>> binaryTreePathSum(TreeNode root, int target) {
        // Write your code here
        return dfs(root,new ArrayList<Integer>(),0,new ArrayList<List<Integer>>(),target);
    }
    public List<List<Integer>> dfs(TreeNode root,List<Integer> node, int sum, List<List<Integer>> paths,int target)
    {
        if(root==null)
        {
            return new ArrayList<List<Integer>>();
        }
        List<List<Integer>> path=new ArrayList<List<Integer>>();
        if(root.left!=null)
        {
            List<Integer> nodes=new ArrayList<Integer>();
        if(node!=null)
        {
            nodes.addAll(node);
        }
            nodes.add(root.val);
            List<List<Integer>> temp=dfs(root.left,nodes,sum+root.val,paths,target);
            if(temp!=null)
            {
             path.addAll(temp);   
            }
        }
        if(root.right!=null)
        {
            List<Integer> nodes=new ArrayList<Integer>();
        if(node!=null)
        {
            nodes.addAll(node);
        }
            nodes.add(root.val);
              List<List<Integer>> temp=dfs(root.right,nodes,sum+root.val,paths,target);
            if(temp!=null)
            {
             path.addAll(temp);   
            }


        }
        if(root.left==null&&root.right==null)
        {
            List<Integer> nodes=new ArrayList<Integer>();
        if(node!=null)
        {
            nodes.addAll(node);
        }
            nodes.add(root.val);
            if(sum+root.val==target)
            {
                path.add(nodes);
            }
            else{
                path=new ArrayList<List<Integer>>();
            }
        }
        return path;
    }
}

原创粉丝点击