NC8 二叉树根节点到叶子节点和为指定值的路径

NC8 二叉树根节点到叶子节点和为指定值的路径

描述
给定一个二叉树和一个值\ sum sum，请找出所有的根节点到叶子节点的节点值之和等于\ sum sum 的路径，
例如：
给出如下的二叉树，\ sum=22 sum=22，

返回
[
[5,4,11,2],
[5,8,9]
]

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @param sum int整型 
     * @return int整型ArrayList<ArrayList<>>
     */
    public ArrayList<ArrayList<Integer>> pathSum (TreeNode root, int sum) {
        // write code here
        ArrayList<ArrayList<Integer>> res = new ArrayList<>() ;
        dfs(root , sum , new ArrayList<>() , res) ;
        return res ;
        
    }
    public void dfs(TreeNode root , int sum , ArrayList<Integer> sub , ArrayList<ArrayList<Integer>> res){
        if(root == null){
            return ;
        }
       // ArrayList<Integer> list = new ArrayList<>(sub) ;
       // list.add(new Integer(root.val)) ;
         sub.add(root.val) ;
        if(root.left == null && root.right == null){
            if(sum == root.val){
                res.add(new ArrayList<Integer>(sub)) ;
            }
           // return ;
        }
       
        dfs(root.left , sum - root.val , sub , res) ;
        dfs(root.right , sum - root.val , sub , res) ;
        sub.remove(sub.size()-1) ;
    }
}