Java计算根节点到叶子节点的所有路径表示的数字之和

 

题目描述

给定一个仅包含数字0−9 的二叉树，每一条从根节点到叶子节点的路径都可以用一个数字表示。
例如根节点到叶子节点的一条路径是1→2→3,那么这条路径就用 123 来代替。
找出根节点到叶子节点的所有路径表示的数字之和
例如：

这颗二叉树一共有两条路径，
根节点到叶子节点的路径 1→2 用数字12 代替
根节点到叶子节点的路径 1→3 用数字13 代替
所以答案为12+13=25

示例1

输入

{1,0}

返回值

10

import java.util.*;

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

public class Solution {
    
    int result = 0;
    /**
     * 
     * @param root TreeNode类 
     * @return int整型
     */
    public int sumNumbers (TreeNode root) {
        
        if(root == null){
            return 0;
        }
        
        preTree(root,0);
        
        return result;
        
    }
    
    public void preTree(TreeNode root,int temp){
        
        if(root == null){
            return;
        }
        
        temp = temp*10 + root.val;
        
        if(root.left == null && root.right == null){
            result += temp;
            return;
        }
        
        preTree(root.left,temp);
        preTree(root.right,temp);
    }
}