LeetCode刷题笔记(树):sum-root-to-leaf-numbers



[toc]

题目描述

Given a binary tree containing digits from0-9only, each root-to-leaf path could represent a number.

An example is the root-to-leaf path1->2->3which represents the number123.

Find the total sum of all root-to-leaf numbers.

For example,

1

/ \
2 3

The root-to-leaf path1->2represents the number12.
The root-to-leaf path1->3represents the number13.

Return the sum = 12 + 13 =25.

给定一个二进制树,只包含0-9的数字,每个根到叶的路径可以代表一个数字。 一个例子是代表数字123的根到叶子路径1-> 2-> 3。 找到所有根到叶子数量的总和。 例如,1 / \ 2 3根到叶子路径1 - > 2表示数字12。 根到叶子路径1-> 3表示数字13。 返回总和= 12 + 13 = 25。

解题思路

两种方法,递归和迭代,递归太简单了,直接贴出代码就不多说了。

前序遍历需要注意的是,因为stack是后入先出规则,所以我们在入栈的时候,先将右子树入栈,再将坐子树入栈,这样在出栈的时候就可以保证是先左后右了。

C++版代码实现

递归

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
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:

void preorder(TreeNode *root, vector<int> &res){
if(root){
res.push_back(root->val);
preorder(root->left, res);
preorder(root->right, res);
}
}

vector<int> preorderTraversal(TreeNode *root) {
vector<int> res;
preorder(root, res);
return res;
}
};

迭代

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
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<int> preorderTraversal(TreeNode *root) {
vector<int> res;
if(root == NULL)
return res;
stack<TreeNode *> sta;
sta.push(root);
while(!sta.empty()){
TreeNode *cur = sta.top();
sta.pop();
res.push_back(cur->val);
if(cur->right)
sta.push(cur->right);
if(cur->left)
sta.push(cur->left);
}
return res;
}
};

系列教程持续发布中,欢迎订阅、关注、收藏、评论、点赞哦~~( ̄▽ ̄~)~

完的汪(∪。∪)。。。zzz

0%