10.2 快速幂算法

最新推荐文章于 2024-08-14 22:49:08 发布

「已注销」

最新推荐文章于 2024-08-14 22:49:08 发布

阅读量146

点赞数 3

文章标签：算法 c++

本文链接：https://blog.csdn.net/scayon/article/details/127144293

版权

快速幂算法

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，在计算x^n同时取模

同余定理

会使用到的同余定理：

(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

什么是快速幂

举个栗子：
3⁵:如果把5看成二进制，即101，那么5=1*2²+0 *2¹+1 *2⁰
则3⁵=3^ (1 *2²+0 *2¹+1 *2⁰)
那么正常需要5次计算而现在只需3次
之前a^b需要运算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;
}