LeetCode129. 求根到叶子节点数字之和

Ember_Sky

于 2020-10-29 14:16:54 发布

阅读量140

点赞数

分类专栏： # LeetCode

本文链接：https://blog.youkuaiyun.com/wzh1378008099/article/details/109358414

版权

LeetCode 专栏收录该内容

111 篇文章

订阅专栏

本文介绍了一种计算二叉树中从根节点到叶子节点的所有数字之和的方法。通过深度优先搜索（DFS），递归地遍历二叉树，并在遍历过程中构建数字。提供了两种实现方案：一种使用辅助数组记录路径，另一种直接在遍历中更新数字。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

//129. 求根到叶子节点数字之和

//使用dfs深度遍历二叉树
//遍历的同时维护一个数组，记录从跟节点到当前节点路径中的数字
//遍历到叶子节点的时候，将数组中的所有数字加到ans上
//注意：不要忘了返回的时候，删除元素
class Solution {
public:
  int ans = 0;
  void dfs(vector<int>& s, TreeNode* node) {
    if (node == NULL) return;
    s.emplace_back(node->val);
    if (node->left == NULL && node->right == NULL) {
      int cnt = 0;
      for (auto x : s) {
        cnt = cnt * 10 + x;
      }
      ans += cnt;
    }
    else {
      dfs(s, node->left);
      dfs(s, node->right);
    }
    s.pop_back();
  }
  int sumNumbers(TreeNode* root) {
    vector<int>s;
    dfs(s, root);
    return ans;
  }
};

//使用dfs，向下传递数字
//遍历到当前节点，就让数字 * 10加上当前节点的值，继续向下传递
//碰到叶子节点，就返回数字
//其他节点就返回dfs左右子节点的和
class Solution {
public:
  int dfs(TreeNode* node, int sum) {
    if (node == NULL) return 0;
    sum = sum * 10 + node->val;
    if (node->left == NULL && node->right == NULL) return sum;
    else return dfs(node->left, sum) + dfs(node->right, sum);
  }
  int sumNumbers(TreeNode* root) {
    return dfs(root, 0);
  }
};