二叉树根节点到叶子节点的所有路径和

二叉树根节点到叶子节点的所有路径和

题目描述

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

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

代码

import java.util.*;

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

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @return int整型
     */
    public int sumNumbers (TreeNode root) {
        // write code here
        if(root == null) return 0;
        //当只有一个root时，返回0，不通过；
        //return method(root.left, root.val) + method(root.right, root.val);
        return method(root, 0);
    }
    
    public int method(TreeNode root, int v){
        if(root == null){
            return 0;
        }
        //当子root都为空时，直接返回
        if(root.left==null && root.right==null){
            return v*10 + root.val;
        }
        //都不为空，都往下探索
        //一个为空，返回为0，另一个继续往下探索
        return  method(root.left, v*10 + root.val) + method(root.right, v*10 + root.val);
    }
}