【leetcode129】求根到叶子节点数字之和【java dfs】

题目

给定一个二叉树，它的每个结点都存放一个 0-9 的数字，每条从根到叶子节点的路径都代表一个数字。

例如，从根到叶子节点路径 1->2->3 代表数字 123。

计算从根到叶子节点生成的所有数字之和。

说明: 叶子节点是指没有子节点的节点。

示例 1:

输入: [1,2,3]
1
/
2 3
输出: 25
解释:
从根到叶子节点路径 1->2 代表数字 12.
从根到叶子节点路径 1->3 代表数字 13.
因此，数字总和 = 12 + 13 = 25.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sum-root-to-leaf-numbers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int sumNumbers(TreeNode root) {
        //边界条件
        if(root==null){
            return 0;
        }
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        List<Integer> path = new ArrayList<Integer>();
        path.add(root.val);
        //dfs
        dfs(res,path,root);
        //计算sum
        int sum = getSum(res);
        return sum;

    }
    void dfs(List<List<Integer>> res,List<Integer> path,TreeNode node){
        //截止条件
        if(node.left==null&&node.right==null){
            res.add(new ArrayList<Integer>(path));
            return;
        }

        //候选结点
        if(node.left!=null){
            path.add(node.left.val);
            dfs(res,path,node.left);
            path.remove(path.size()-1);
        }

        if(node.right!=null){
            path.add(node.right.val);
            dfs(res,path,node.right);
            path.remove(path.size()-1);
        }

    }
    int getSum(List<List<Integer>> res){
        int sum = 0;
        for(List<Integer> path:res){
            int num = 0;
            for(Integer a:path){
                num = num*10+a;
            }
            sum+=num;
        }
        return sum;
    }


}