题目链接

题目描述:

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

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

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

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

实例:
129求根到叶子节点数字之和 - 图1

解题思路

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

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

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

  1. class Solution {
  2. public:
  3. int sumNumbers(TreeNode* root) {
  4. if(!root)
  5. return sum;
  6. helper(root, root->val); //提前赋值,后面递归的赋值也要相应改变
  7. return sum;
  8. }
  9. void helper(TreeNode* root, int cur){ //cur表示每一条路径和
  10. if(!root)
  11. return;
  12. if(!root->left && !root->right){
  13. sum += cur;
  14. return;
  15. }
  16. if(root->left)
  17. helper(root->left, cur * 10 + root->left->val);
  18. if(root->right)
  19. helper(root->right, cur * 10 + root->right->val);
  20. return;
  21. }
  22. private:
  23. int sum = 0; //记录所有和
  24. };
  1. class Solution {
  2. public:
  3. int sumNumbers(TreeNode* root) {
  4. if(!root)
  5. return sum;
  6. helper(root, to_string(root->val));
  7. return sum;
  8. }
  9. void helper(TreeNode* root, string cur){
  10. if(!root)
  11. return;
  12. if(!root->left && !root->right){
  13. //字符串转int
  14. stringstream ss;
  15. int curNum;
  16. ss << cur;
  17. ss >> curNum;
  18. sum += curNum;
  19. return;
  20. }
  21. if(root->left)
  22. helper(root->left, cur + to_string(root->left->val));
  23. if(root->right)
  24. helper(root->right, cur + to_string(root->right->val));
  25. return;
  26. }
  27. private:
  28. int sum = 0;
  29. };

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