JZ049从根节点到叶子节点的数字之和

title: 从根节点到叶子节点的数字之和

📃 题目描述

题目链接：从根节点到叶子节点的数字之和、相同题目

🔔 解题思路

方法一：递归回溯，主要判断好：当前结点传什么值进来，我传入的是上一个节点加入节点值后扩大十倍；

class Solution {
public:
int res = 0;
int sumNumbers(TreeNode* root) {
backTracking(root, 0);
return res;
}
void backTracking(TreeNode *root, int num) {
if (!root->left && !root->right) {
res += (num + root->val);
return;
}

num = (num + root->val) * 10;//传入扩大十倍的值；
if (root->left) {
backTracking(root->left, num);
}
if (root->right) {
backTracking(root->right, num);
}
}
};

方法二：迭代深度优先（用栈），一个栈保存节点，另一个栈保存到当前节点的路径值，比如前面经过数值1，经过数值2，才到当前节点数值4，那么当前节点的路径值为124；

当然也可以用广度优先搜索（有队列）

class Solution {
public:
int sumNumbers(TreeNode* root) {
stack<TreeNode*> treeSt;//保存节点指针
stack<int> sumSt; //保存节点的路径值
treeSt.push(root);
sumSt.push(root->val);
int res = 0;
while (!treeSt.empty()) {
TreeNode *cur = treeSt.top();
treeSt.pop();
int nodeSum = sumSt.top();
sumSt.pop();

if (!cur->left && !cur->right) {
res += nodeSum;
}

if (cur->left) {
treeSt.push(cur->left);
sumSt.push(nodeSum * 10 + cur->left->val);
}
if (cur->right) {
treeSt.push(cur->right);
sumSt.push(nodeSum * 10 + cur->right->val);
}
}
return res;
}
};

💥 复杂度分析

• 时间复杂度：o(n);
• 空间复杂度：O(n);