【多次过】Lintcode 376. 二叉树的路径和

 

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

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

样例

给定一个二叉树，和  目标值 = 5：

     1
    / \
   2   4
  / \
 2   3

返回：

[
  [1, 2, 2],
  [1, 4]
]

解题思路：

    此题同样有个与Lintcode 155.二叉树的最小深度一样的陷阱，递归到底的情况，一个有效路径的定义是从根节点到任何一个叶子节点。所以当满足目标条件时还要判断最后一个结点的左右子树是否为空。

/**
 * 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
        List<List<Integer>> res = new ArrayList<>();
        List<Integer> path = new ArrayList<>();
        int sum = 0;
        
        getSum(root, target, sum, path, res);
        
        return res;
    }
    
    private void getSum(TreeNode root, int target, int sum, List<Integer> path, List<List<Integer>> res){
        if(root == null)
            return;
        
        sum += root.val;
        path.add(root.val);
        
        //若当前节点为叶子节点且满足插入条件
        if(root.left == null && root.right == null && sum == target){
            res.add(new ArrayList<Integer>(path));
            path.remove(path.size() - 1);
            return;
        }else{
            getSum(root.left, target, sum, path, res);
            getSum(root.right, target, sum, path, res);
            
            path.remove(path.size() - 1);
        }
    }
}