考察知识点: 深度优先搜索、先序遍历
题目分析:
由于每一个产奶量都是由一条从根节点
到叶子节点
的路径构成的,每一个数的高位都是根节点,低位是叶子节点,即产奶量从高位到低位对应着二叉树的深度从低到高。可见我们可以使用先序遍历
的方式来解决问题。
在遍历时,首先要将之前得到的产奶量数字末尾加上该节点的数字,可以先将原来的数字乘以10,再加上该节点的数字。当遍历到叶子节点
,即左右孩子均为空节点时,将num加入到sum中。由于num是值传递,遍历左子树时修改的num不会影响到右子树。
所用编程语言: C++
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @return int整型
*/
void traverse(TreeNode *root, int num, int &sum) {
if (!root) return;
num = num * 10 + root->val;
if (!root->left && !root->right) {
sum += num;
}
traverse(root->left, num, sum);
traverse(root->right, num, sum);
}
int sumNumbers(TreeNode* root) {
// write code here
if (!root) return 0;
int sum = 0;
traverse(root, 0, sum);
return sum;
}
};