LeetCode Sum Root to Leaf Numbers

Dylan_Java_NYC

于 2015-09-07 22:29:32 发布

阅读量345

点赞数

分类专栏： Recursion Tree DFS LeetCode

本文链接：https://blog.youkuaiyun.com/Dylan_Java_NYC/article/details/48274805

版权

LeetCode 同时被 3 个专栏收录

122 篇文章

订阅专栏

Tree

32 篇文章

订阅专栏

Recursion

30 篇文章

订阅专栏

本文介绍了一种解决LeetCode上Sum Root to Leaf Numbers问题的方法。通过递归遍历二叉树，当到达叶子节点时累加路径上的数值。文章详细解释了递归函数的设计思路及实现细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题链接在这里：https://leetcode.com/problems/sum-root-to-leaf-numbers/

本题与Path Sum II相似，都是递归调用。终止条件都是左右child都是空，否则cur*10加child.val, 递归调用函数，再去掉尾节点。

Note: 1. helper 是pass by value, 所以res一定要用个array存储。

2. 递归调用后需要去掉尾节点，本题中除以10即可。

3. root == null 的corner case 是在主函数里单独讨论，helper里不需要重复讨论。

AC Java:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public int sumNumbers(TreeNode root) {
        if(root == null){
            return 0;
        }
        int cur = root.val;
        int [] res = new int[1];
        res[0] = 0;
        helper(root, res, cur);
        return res[0];
    }
    private void helper(TreeNode root, int [] res, int cur){
        if(root.left == null && root.right == null){
            res[0] += cur;
            return;
        }
        if(root.left != null){
            cur*=10;
            cur+=root.left.val;
            helper(root.left,res,cur);
            cur/=10;
        }
        if(root.right != null){
            cur*=10;
            cur += root.right.val;
            helper(root.right, res, cur);
            cur/=10;
        }
    }
}