leetcode129 Sum Root to Leaf Numbers java 算法

来源:互联网 发布:windows 8.1 激活 编辑:程序博客网 时间:2024/04/29 16:13

题目描述:计算从根节点到叶节点的路径构成的数字的和。

算法描述:递归

代码如下:

public static int  sumNumbers(TreeNode root) {
        if(root==null){
            return 0;
        }
        List<List<Integer>> result = new ArrayList<List<Integer>>();//这两个list可以帮助我们查看从根节点到叶节点有哪些路径
        List<Integer> ls = new ArrayList<Integer>();
        int sum = 0;
        return (findAll(root,result,ls,sum,0));
    }
    public static int  findAll(TreeNode root,List<List<Integer>> result,List<Integer> ls,int sum,int cur){
        if(root.left==null && root.right==null){
            ls.add(root.val);
            int tt = cur;
            cur = cur*10+root.val;
            List<Integer> temp = new ArrayList<Integer>(ls);
            result.add(temp);
            System.out.println(cur);
            sum = sum+cur;
            ls.remove(ls.size()-1);
            return cur;
        }
        int sum1 = 0;
        int sum2 = 0;
        if(root.left!=null){
            ls.add(root.val);
            sum1 = findAll(root.left,result,ls,sum,cur*10+root.val);
            ls.remove(ls.size()-1);
        }
        if(root.right!=null){
            ls.add(root.val);
            sum2 = sum + findAll(root.right,result,ls,sum,cur*10+root.val);
            ls.remove(ls.size()-1);
        }
        return sum1+sum2;
    }

0 0