第7天,为什么今天随机到了一道水题。。。
今天的题目是129. Sum Root to Leaf Numbers
恩,这是一道打卡水题,明明是道Easy
的题目,却混进了Medium
中:
这道题有个比较容易想错的方法就是,用后序遍历做:
1int sumNumbers(TreeNode* root) {2 if (root == nullptr) return 0;3 return sumNumbers(root->left) + sumNumbers(root->right) + 20 * root->val;4}
这个方法对满二叉树是没问题的,但是对于普通的二叉树就会有问题,如:
1 12 /3 2
也就是说,在两边高度不一样时,是会出问题的,为了解决这个问题,我们要把思路换过来,把root
到当前节点的值计算出来,并传给子节点去计算:
1class Solution {2public: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};