题目描述
给出一棵二叉树,其上每个结点的值都是 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
。
解题思路:
思路并不难,就是采用dfs遍历二叉树,将所有的二进制串表示的整数求模之后返回其和。
我遇到的问题主要是当这个二进制串表示的整数非常大的时候,计算机无法表示了,那么就不能等把这个数字表示出来再进行求模,而必须在计算的过程中求模!
如果你也遇到这个问题,具体方法可参考二进制数转整数、整数转二进制数、二进制数求模里面的第三部分——二进制求模,里面有详细讲解求一个二进制串表示的大整数对另一个整数求模的结果,结果用十进制整数返回。相信会给你很多启发!
AC代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
#define MOD 1000000007
class Solution {
public:
int res;
void dfs(TreeNode* T, int curNum)
{
if(T->left)
dfs(T->left, (curNum * 2 + T->left->val) % MOD);
if(T->right)
dfs(T->right, (curNum * 2 + T->right->val) % MOD);
if(!T->left && !T->right)
(res += curNum % MOD) %= MOD;
}
int sumRootToLeaf(TreeNode* root) {
res = 0;
dfs(root, root->val);
return res;
}
};