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);
        }
    }
}