avatar

目录
129求根到叶子节点数字之和

题目链接

题目描述:

给定一个二叉树,它的每个结点都存放一个 0-9 的数字,每条从根到叶子节点的路径都代表一个数字。

例如,从根到叶子节点路径 1->2->3 代表数字 123。

计算从根到叶子节点生成的所有数字之和。

说明: 叶子节点是指没有子节点的节点。

实例:
leetcode

解题思路

  与#257思路类似,套用先序遍历框架,将焦点移到对当前结点的操作和对传入参数对设置,剩下的交给递归。

  1. 判断当前结点是否为叶子结点。如果是,显然意味着一趟路径已然完成,将这一趟的路径和传给总数和sum。之后return回上一层(叶子结点的父结点),换路继续递归找出其与路径和(有回溯的思想在其中)。
  2. 既然有回溯的思想,那显然会出现状态重置的部分。而且根据本题题意,需要设置暂时存储的变量用于存储每一条路径和。对于该问题,可以通过添加辅助函数,扩展一个参数cur,用于暂时存储每一条路径和,然后在其传入参数的赋值部分,由于一开始就赋上当前结点的值,因此后面递归的赋值也要赋上左右子节点的值。注意:递归蕴含着栈的思想,每个结点,只有当与它相关的左右子树全部遍历完成才会返回它的父结点,也就是说返回之前它自己的cur是一直存在的。

乘10自增,执行用时:4ms

cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
class Solution {
public:
int sumNumbers(TreeNode* root) {
if(!root)
return sum;

helper(root, root->val); //提前赋值,后面递归的赋值也要相应改变
return sum;
}

void helper(TreeNode* root, int cur){ //cur表示每一条路径和
if(!root)
return;

if(!root->left && !root->right){
sum += cur;
return;
}

if(root->left)
helper(root->left, cur * 10 + root->left->val);
if(root->right)
helper(root->right, cur * 10 + root->right->val);

return;
}

private:
int sum = 0; //记录所有和
};

### 字符串转int,执行用时:8ms
cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
class Solution {
public:
int sumNumbers(TreeNode* root) {
if(!root)
return sum;

helper(root, to_string(root->val));
return sum;
}

void helper(TreeNode* root, string cur){
if(!root)
return;

if(!root->left && !root->right){
//字符串转int
stringstream ss;
int curNum;
ss << cur;
ss >> curNum;

sum += curNum;
return;
}

if(root->left)
helper(root->left, cur + to_string(root->left->val));
if(root->right)
helper(root->right, cur + to_string(root->right->val));

return;
}

private:
int sum = 0;
};

如果有错误或者不严谨的地方,请务必给予指正,十分感谢。

文章作者: breadhunter
文章链接: http://yoursite.com/2020/03/09/129%E6%B1%82%E6%A0%B9%E5%88%B0%E5%8F%B6%E5%AD%90%E8%8A%82%E7%82%B9%E6%95%B0%E5%AD%97%E4%B9%8B%E5%92%8C/
版权声明: 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 breadhunter
打赏
  • 微信
    微信
  • 支付寶
    支付寶