LintCode | 376. 二叉树的路径和

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


题目不难,关键在于想清楚递归怎么判断。

/**
* 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) {
        List<List<Integer>> list = new ArrayList<List<Integer>>();
        if(root != null) {
            target = target - root.val;
            List<Integer> way = new ArrayList<Integer>();
            way.add(root.val);
            list.add(way);
            findWay(list, way, root, target);
        }
        return list;
    }

    private void findWay(List<List<Integer>> list, List<Integer> way, TreeNode node, int target) {
        if(node.left != null && node.right != null) {
            //当左右子树都存在时,则用API中addAll方法创建当前路径的一个路径副本,供右子树走
            List<Integer> copy = new ArrayList<Integer>();
            copy.addAll(way);
            list.add(copy);
            way.add(node.left.val);
            findWay(list, way, node.left, target - node.left.val);
            copy.add(node.right.val);
            findWay(list, copy, node.right, target - node.right.val);
        } else if(node.left != null && node.right == null) {
            //当左子树或者右子树存在时,则无需创建副本,继续在当前路径添加即可
            way.add(node.left.val);
            findWay(list, way, node.left, target - node.left.val);
        } else if(node.left == null && node.right != null) {
            way.add(node.right.val);
            findWay(list, way, node.right, target - node.right.val);
        } else {
            //当左右子树都为空时,说明已到叶结点,通过判断target是否已减到0来判断路径和是否为原target
            if(target != 0) list.remove(way);
        }
    }
}
发布了84 篇原创文章 · 获赞 0 · 访问量 3万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章