376 - 二叉树的路径和

最新推荐文章于 2019-03-10 01:32:17 发布

原创最新推荐文章于 2019-03-10 01:32:17 发布 · 364 阅读

0 ·

CC 4.0 BY-SA版权

lintcode 专栏收录该内容

218 篇文章

订阅专栏

本文介绍了一种使用递归方法寻找二叉树中所有从根节点到叶子节点的路径，并计算这些路径上的数值之和，最终筛选出和为目标值的路径。

2017.9.28

先用递归的办法找到所有的根节点到叶子节点的路径。

再一一计算和。

/**
 * 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>> pathToLeaf(TreeNode root){
		List<List<Integer>> res = new LinkedList<>();
		if(root == null){
			return res;
		}
		if(root.left == null && root.right == null){
			List<Integer> tmp = new LinkedList<Integer>();
			tmp.add(root.val);
			res.add(tmp);
			return res;
		}
		if(root.left != null){
			List<List<Integer>> resLeft = pathToLeaf(root.left);
			for(List<Integer> list: resLeft){
				List<Integer> tmp = new LinkedList<Integer>();
				tmp.add(root.val);
				tmp.addAll(list);
				res.add(tmp);
			}
		}
		if(root.right != null){
			List<List<Integer>> resRight = pathToLeaf(root.right);
			for(List<Integer> list: resRight){
				List<Integer> tmp = new LinkedList<Integer>();
				tmp.add(root.val);
				tmp.addAll(list);
				res.add(tmp);
			}
		}
		return res;
	}
	
	public List<List<Integer>> binaryTreePathSum(TreeNode root, int target) {
        // write your code here
		List<List<Integer>> res = new LinkedList<>();
		if(root == null){
			return res;
		}
		List<List<Integer>> path = pathToLeaf(root);
		for(List<Integer> list : path){
		    //System.out.println(list);
			int sum = 0;
			for(Integer num : list){
				sum = sum + num;
			}
			if(sum == target){
				res.add(list);
			}
		}
		
		return res;
		
    }
}