Code & Func
2019-03-06

第7天,为什么今天随机到了一道水题。。。

今天的题目是129. Sum Root to Leaf Numbers

恩,这是一道打卡水题,明明是道Easy的题目,却混进了Medium中:

这道题有个比较容易想错的方法就是,用后序遍历做:

1
int sumNumbers(TreeNode* root) {
2
if (root == nullptr) return 0;
3
return sumNumbers(root->left) + sumNumbers(root->right) + 20 * root->val;
4
}

这个方法对满二叉树是没问题的,但是对于普通的二叉树就会有问题,如:

1
1
2
/
3
2

也就是说,在两边高度不一样时,是会出问题的,为了解决这个问题,我们要把思路换过来,把root到当前节点的值计算出来,并传给子节点去计算:

1
class Solution {
2
public:
3
int sumNumbers(TreeNode* root) {
4
return sumNumbers(root, 0);
5
}
6
int sumNumbers(TreeNode* root, int v) {
7
if (root == nullptr) return 0;
8
9
v = v*10 + root->val;
10
if (root->right == nullptr && root->left == nullptr) return v;
11
12
return sumNumbers(root->left, v) + sumNumbers(root->right, v);
13
}
14
};
上一条动态