题目描述
给出一棵二叉树,其上每个结点的值都是 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 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-of-root-to-leaf-binary-numbers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
class Solution {
public:
int ans;
int mod;
stack<int> sk;
void dfs(TreeNode* root){
if(root == NULL) return;
if(root->left == NULL && root->right==NULL){
sk.push(root->val);
stack<int> tsk = sk;
int i = 0,tans = 0;
while(!tsk.empty()){
tans += tsk.top()*pow(2,i);
tans%=mod;
tsk.pop();
i++;
}
ans += tans;
ans%=mod;
sk.pop();
return;
}
sk.push(root->val);
dfs(root->left);
dfs(root->right);
sk.pop();
}
int sumRootToLeaf(TreeNode* root) {
ans = 0;mod = 1e9+7;
dfs(root);
return ans;
}
};