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

二叉树？生成数字之和！

题目
给定一个二叉树，它的每个结点都存放一个 0-9 的数字，每条从根到叶子节点的路径都代表一个数字。
例如，从根到叶子节点路径 1->2->3 代表数字 123。
计算从根到叶子节点生成的所有数字之和。
说明: 叶子节点是指没有子节点的节点。

例子
示例 1:
输入: [1,2,3]

  1 
 / \
2   3

输出: 25
解释:
从根到叶子节点路径 1->2 代表数字 12.
从根到叶子节点路径 1->3 代表数字 13.
因此，数字总和 = 12 + 13 = 25.

输入: [4,9,0,5,1]

    4
   / \
  9   0
 / \
5   1

输出: 1026
解释:
从根到叶子节点路径 4->9->5 代表数字 495.
从根到叶子节点路径 4->9->1 代表数字 491.
从根到叶子节点路径 4->0 代表数字 40.
因此，数字总和 = 495 + 491 + 40 = 1026.

题目分析

• 目标：路径结点组成数字 ⇒ 求和

1. 寻找从根结点到叶子的路径 ⇒ 路径上的结点排列组成数字
2. 数字求和

解题思路

变量	作用
find()	寻找路径函数
transform()	路径转化为整数函数

过程

1. 当前结点加入当前路径 ⇒ 更新当前路径和
2. 如果为叶子结点 ⇒ 说明找到一条路径 ⇒ 添加到路径集合
3. 如果左子树不为空 ⇒ 更新左子树为当前路径 重复2,3
4. 如果右子树不为空 ⇒ 更新右子树为当前路径 重复2,3

代码如下

void find(TreeNode*root, vector<vector<int>>&num, vector<int>path)
 {
     path.push_back(root->val);                                   //当前结点加入当前路径
     if(!root->left && !root->right) num.push_back(path);         //如果为叶子结点就添加到路径集合
     if(root->left) find(root->left, num, path);                  //左子树递归
     if(root->right) find(root->right, num, path);                //右子树递归
 }
 int transform(vector<int> k)
 {
     int cnt;
     int sum = 0;
     for(int i = 0; i < k.size(); i++)
     {
         sum = sum*10 + k[i];
     }
     return sum;
 }
class Solution {
public:
    int sumNumbers(TreeNode* root) {
        if(!root) return 0;
        vector<vector<int>> num;
        vector<int>path;
        int sum = 0;
        find(root, num, path);
        for(int i = 0; i < num.size(); i++)
        {
            sum += transform(num[i]);                  //将路径转化为整数求和
        }
        return sum;
    }
};