[LeetCode 129] - 根节点到叶子节点数字求和(Sum Root to Leaf Numbers)

问题

给定一个节点取值只包含数字0-9的二叉树，每条根节点至叶子节点的路径都可以表示一个数字。

例如，根至叶子路径1->2->3表示数字123。

求所有根至叶子数字的和。

例如，

1

/ \

2 3

根至叶子路径1->2表示数字12

根至叶子路径1->3表示数字13

返回值sum=12+13=25

初始思路

要求和就必须把所有路径列出来，所以解决问题的核心就是要遍历所有根到叶子的路径。遍历二叉树容易想到通过递归访问左右子节点直到找到叶子，此时发现一条路径。所以递归结束的条件是当前节点为叶子节点，而一个叶子节点是没有子节点的，所以递归结束条件用代码表示为：root->left == 0 && root->right == 0

一次递归结束时，需要知道路径中都包含了什么值以求出根至叶子数，因此我们需要一个容器记录本次都走过了哪些节点，这里使用一个std::list<int>。在递归访问过程中，每访问到一个节点，我们把当前节点的值放入容器尾部，而函数返回前把该值弹出以表示返回上一个节点。得到了一个路径中所有节点的值后就可以将其转为转为数字了。由于值都是按顺序放入容器的，所以从头到尾对容器中的每一个值执行 和 = 10 * 当前位置的值 + 和 的逻辑就可以算出该数字了。算出一条路径的数字后我们再将其累加到总和即可。

把上面的理论写成代码：

class Solution {
public:
int sumNumbers(TreeNode *root)
{
sum_ = 0;
path_.clear();

if(root)
{
DoSum(root);
}

return sum_;
}

private:
void DoSum(TreeNode *root)
{
path_.push_back(root->val);
if(root->left == 0 && root->right == 0)
{
CountOnePath();
}
else
{
if(root->left)
{
DoSum(root->left);
}
if(root->right)
{
DoSum(root->right);
}
}

path_.pop_back();
}

void CountOnePath()
{
int sum = 0;
for(auto iter = path_.begin(); iter != path_.end(); ++iter)
{
sum = 10 * sum + *iter;
}

sum_ += sum;
}

int sum_;
std::list<int> path_;
};

提交后一次性Judge Small和Judge Large通过。看来本题并没有什么机关。