*LeetCode-Path Sum II

1024 ワード

以前やってたbacktrackingに似てるけど今回addのnodeを削除するのに注意!!さもなくば左recurseはleftをプラスして右に着いてremove leftがなくて続けました
public class Solution {
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        List<List<Integer>> res = new ArrayList<List<Integer>> ();
        List<Integer> list = new ArrayList<Integer> ();
        helper ( res, list, root, sum );
        return res;
    }
    public void helper ( List<List<Integer>> res, List<Integer> list, TreeNode root, int sum ){
        if ( root == null )
            return;
        list.add ( root.val );
        if ( root.val == sum && root.left == null && root.right == null ){
            res.add ( new LinkedList (list) );
            list.remove( list.size() - 1 );
            return;
        }
        helper ( res, list, root.left, sum - root.val );
        helper ( res, list, root.right, sum - root.val );
        list.remove( list.size() - 1 );
    }
}