Leetcode 129 Sum Root to leaf Numbers

sure00

已于 2022-01-25 11:32:31 修改

阅读量4.5k

点赞数

文章标签： leetcode 算法职场和发展

于 2022-01-25 11:20:00 首次发布

本文链接：https://blog.csdn.net/sure00/article/details/122679282

版权

本文介绍了一种计算二叉树所有从根节点到叶子节点路径数值之和的算法。通过递归方法和队列迭代两种方式实现，递归方法通过深度优先搜索更新路径数值并累加到结果中；队列迭代方法则采用广度优先搜索策略，更新每个节点的数值直至到达叶子节点。

摘要由CSDN通过智能技术生成

Input: root = [1,2,3]
Output: 25
Explanation:
The root-to-leaf path 1->2 represents the number 12.
The root-to-leaf path 1->3 represents the number 13.
Therefore, sum = 12 + 13 = 25.

题目理解不难，方法一也很固定。递归遍历整个树的node。

并且动态更新number数值。 val*10+root->val

当到达叶子节点时候把最新的val*10+root->val 加到结果的ret里面。

然后就这样遍历所有的叶子节点。

/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int ret=0;
void help(TreeNode* root, int val) {
if(root->left == nullptr && root->right == nullptr) {
val = val*10 + root->val;
ret += val;
return;
}

if(root->left) {
help(root->left, val*10 + root->val);
}

if(root->right) {
help(root->right, val*10 + root->val);
}

return;
}

int sumNumbers(TreeNode* root) {
if(root == nullptr)
return 0;

help(root, 0);

return ret;
}
};

方法二：可以用队列存储每一层的节点，然后出队列的同时push 进去新的子孩子。不过在push 子孩子进fifo之前，要更新子孩子的value: root *10 +孩子的数值。一层层的做下去，到了叶子节点就是最后要的数值。

while(size--) {
root=myq.front();

myq.pop();
if(root->left) {
root->left->val += root->val * 10;
myq.push(root->left);
}
if(root->right) {
root->right->val += root->val * 10;
myq.push(root->right);
}
if(root->left == nullptr && root->right == nullptr)
ret +=root->val;
}
}
return ret;
}
};