LintCode 1353. 根节点到叶节点求和(DFS)

2020-07-13 16:48:17 浏览数 (1)

1. 题目

给定仅包含来自0-9的数字的二叉树,每个根到叶路径可以表示数字。 举个例子:root-to-leaf路径1-> 2-> 3,它代表数字123,找到所有根到叶的数的总和

代码语言:javascript复制
样例1
输入: {1,2,3}
输出: 25
解释:
    1
   / 
  2   3
路径 1->2 表示数字 12.
路径 1->3 表示数字 13.
因此, sum = 12   13 = 25.

样例2
输入: {4,9,0,5,1}
输出: 1026
解释:
    4
   / 
  9   0
 / 
5   1
路径 4->9->5 表示数字 495.
路径 4->9->1 表示数字 491.
路径 4->0 表示数字 40.
因此, sum = 495   491   40 = 1026.

注意事项
叶节点是没有子节点的节点

2. 解题

代码语言:javascript复制
class Solution {
    int sum = 0;
public:
    int sumNumbers(TreeNode * root) {
        if(!root)
            return 0;
        dfs(root,0);
        return sum;
    }

    void dfs(TreeNode* root, int s)
    {
        if(!root)
            return;
        s = 10*s root->val;
        dfs(root->left,s);
        dfs(root->right,s);
        if(!root->left && !root->right)//是叶节点
            sum  = s;
    }
};

100% 数据通过测试 总耗时 50 ms 您的提交打败了 48.79% 的提交!

0 人点赞