问题描述
快速幂其实就是求解a^b%p的过程,其中a,b,p都是int可以表示的非负数集合,我们如何快速得到该结果呢?如果我们采用暴力的方式求解该问题,时间复杂度是比较高的, 它与b呈线性关系。所以当b的数目很大时我们需要找到一种求解幂的快速方法。
解决方案
显然如果我们将b分解为一些数的和,我们就可以分别求解各个数下,a的指数幂,再将所有的指数幂相乘即可。而一种很好的分解方式就是将b分解为2的指数幂的和,这样我们就可以根据2的指数幂从小到大的过程快速求解的到a^b%p的值。
code
#include<bits/stdc++.h>
using namespace std;
int pow(int a,int b,int p){
int ans=1%p;//非常神奇的一点,可以避免b=0的初始化
for(;b;b>>=1){
if(b&1)ans=(long long)ans*a%p;//只有b的2的指数幂分解中,相应位置上的数字为1,才可以求解指数幂
a=(long long)a*a%p;//求解pow(2,i)%p的值,实际上是以从小到大递推的方式求解的。
}
return ans;
}
int main(){
int a,b,p;
cin>>a>>b>>p;
cout<<pow(a,b,p)<<endl;
return 0;
}