WA的原因
题目:已知长度为n的数字串s(下标从1开始),有q次询问,每次询问给出l和r,问[l,r]区间构成的数字对7取模之后的结果是多少。(不保证没有前导零)
input
第一行输入n,q(1≤n,q≤2∗10^5),表示数字串的长度和询问次数。
第二行,输入数字串s。
接下来q行,每行两个数字l,r(1≤l≤r≤n),表示询问区间。
output
共q行,每行一个数字表示答案。
sample input
6 2
102345
1 2
2 4
sample output
1
5
第一份代码(WA)
int main(){
int n,q,l,r,ans;
scanf("%d%d",&n,&q);
s[0]=0,a[0]=0;
getchar();
for(int i=1;i<=n;i++){ //输入a[i]后计算当前数的十进制数值
scanf("%c",&a[i]);
s[i]=s[i-1]+(a[i]-‘0’)*pow(10,(n-i)); //问题出在这里:pow适用于浮点数 用于整型会丢精度,而且当n为10^5时 10的1e5次方...会爆炸的
}
for(int i=1;i<=q;i++){
scanf("%d%d",&l,&r);
ans=(s[r]-s[l-1])/pow(10,(n-r)); //计算截取区间的十进制数值 再整体取模(会爆炸 !)
printf("%d\n",ans%7);
}
return 0;
}
所以就需要快速幂来替换pow,在计算xn同时取模
同余定理
- (a+b)%mod=(a%mod+b%mod)%mod
- (a-b)%mod=(a%mod-b%mod+mod)%mod
- (a* b)%mod=(a%mod*b%mod)%mod
什么是快速幂
举个栗子:
35:如果把5看成二进制,即101,那么5=1*22+0 *21+1 *20
则35=3^ (1 *22+0 *21+1 *20)
那么正常需要5次计算而现在只需3次
之前ab需要运算b次 现在只需log2(b)次 算法复杂度O(log2(b))
ll qpow(lla,llb){
ll ans=1;
while(b){
if(b&1){
ans*=a;
}
a*=a;
b>>=1;
}
return ans;
}
例题的AC代码
int qpow(int a,int n,int mod){ //底数a,指数n,模数mod
int ans=1;
while(n>0){
if(n&1){ //计算时直接取模,防止总数值过大
ans=ans*a%mod;
}
a=a*a%mod;
n>>=1;
}
return ans;
}
int main(){
int n,q,left,right,res; //数字串长度n,询问次数q,取出数字的区间[left,right]
scanf("%d%d",&n,&q);
getchar();
s[0]=0;
for(int i=1;i<=n;i++){ //从1到n遍历,求前缀和
scanf("%c",&a[i]);
s[i]=(s[i-1]*10+(a[i]-'0'))%7;
}
for(int i=1;i<=q;i++){
scanf("%d%d",&left,&right);
res=s[right]+7-(s[left-1]*qpow(10,right-left+1,7))%7; //同余定理
printf("%d\n",res%7);
}
return 0;
}