给出一棵二叉树,其上每个结点的值都是 0 或 1 。每一条从根到叶的路径都代表一个从最高有效位开始的二进制数。例如,如果路径为 0 -> 1 -> 1 -> 0 -> 1,那么它表示二进制数 01101,也就是 13 。
对树上的每一片叶子,我们都要找出从根到该叶子的路径所表示的数字。
以 10^9 + 7 为模,返回这些数字之和。
示例:
输入:[1,0,1,0,1,0,1]
输出:22
解释:(100) + (101) + (110) + (111) = 4 + 5 + 6 + 7 = 22
提示:
树中的结点数介于 1 和 1000 之间。
node.val 为 0 或 1 。
解:
思路:先序遍历二叉树,定义一个变量totalSum用于保存各个路径所表示的数的总和,定义一个变量currentSum代表遍历到当前节点时的值。当且仅当节点为叶子节点时,将currentSum的值追加到totalSum中。这里totalSum传入的是一个指针,而currentSum是一份拷贝(在递归过程中,不同节点的currentSum是不同的)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
#ifndef
M
#define M
1000000007
#endif
void
getSum(
struct
TreeNode* root,
int
currentSum,
int
*totalSum){
if
(root==NULL)
return
;
currentSum=currentSum*
2
+root->val;
if
(root->left==NULL && root->right==NULL) *totalSum+=currentSum;
getSum(root->left,currentSum,totalSum);
getSum(root->right,currentSum,totalSum);
}
int
sumRootToLeaf(
struct
TreeNode* root){
if
(root==NULL)
return
0
;
int
currentSum=
0
,totalSum=
0
;
getSum(root,currentSum,&totalSum);
return
totalSum%M;
}