算法介绍
假设我们要求的数为a^b,那么我们可以先得到a^(b/2),而a^(b/2)又可以通过a^(b/4)得到,从而在O(logn)时间内完成运算。
非递归的方法为:假设为2^23,将23转化为2进制(10111),即为2^16 * 2^4 * 2^2 * 2,从而得出每当b&1的结果为1时就把当前的倍数赋值给ans,每次循环a都翻倍。
注意:如果进行模除,则a(翻倍后)和ans都要进行模除。
算法模拟
a=2,b=23(10111),ans=1
第一次:a=2^1,b&1=1,说明需要乘以a,ans*=a b右移一位变成(1011)
第二次:a=2^2,b&1=1,说明需要乘以a,ans*=a b右移一位变成(101)
第二次:a=2^4,b&1=1,说明需要乘以a,ans*=a b右移一位变成(10)
第二次:a=2^8,b&1=0,说明不需要乘以a,ans不变, b右移一位变成(1)
第二次:a=2^16,b&1=1,说明需要乘以a,ans*=a b右移一位变成(0),运行结束
测试代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<string>
#include<algorithm>
#define INF 99999999
using namespace std;
int fun(int a,int b,int m){//(a^b)%m的快速幂实现
long long int ans=1;
long long int ta=a;
while(b){
if(b&1)ans=(ans*ta)%m;
ta=(ta*ta)%m;
b>>=1;
}
return ans;
}
int main(){
printf("%d\n",fun(2,10,1000));
return 0;
}