快速幂
通常情况求,我们求 ab都习惯按照如下方式做:
#include<iostream>
#include<math.h>
using namespace std;
int myPow(int a,int b){
int res=1;
for(int i=0;i<b;i++){
res*=a;
}
return res;
}
int main(){
int a,b;
cin>>a>>b;
cout<<pow(a,b)<<endl;//直接调用pow()函数
cout<<myPow(a,b)<<endl;//自定义myPow()函数
}
但是,这样做,在做比较大的求幂时,可能花的时间会比较多,我们有没有更快的方法呢?当然有,我们接着往下看。
比如在求210,我们可以转换成求25*25的形式,这样就比直接求10次快。快速幂就是用的这种思路。
我们都知道,数字以二进制的方式存储在计算机中,比如10=(1010)2,所以我们在求幂的时候,可以考二进制。我们看代码
#include<iostream>
using namespace std;
int fast_pow(int a,int b){
int res=1;
while(b){
if(b&1){ //b的二进制最低位与1算数与。
res=res*a;
}
b>>=1; //b右移一位。
a=a*a; //b每右移一位,a就翻一倍。
/*如2的10次幂 10=(1010)2
*1)0&1=0; b=101;a=a*a=4;
*2)1&1=1;res=res*a=4;b=10;a=a*a=16;
*3)0&1=0;b=1;a=a*a=256;
*4)1&1=1;res=res*a=1024;b=0(退出循环)。
*/
}
return res;
}
int main(){
int a,b;
cin>>a>>b;
cout<<fast_pow(a,b)<<endl;
}
从这上面的例子,我们可以看出快速幂能使求幂的的效率高不少。接下来我们看一个快速幂的例子:
原题链接:XDOJ1027
题目:
北雷老中医不懂 Fibonacci,于是他定义了一种名叫Feibonaqi的数列,数列满足以下要求:
1.f(0)=0,f(1)=1
2.f(n)=2f(n-1)+f(n-2),n>1
那么问题来了,这个数列的第n项是多少呢?
(不为难大家用高精度了,你只需要将结果对1,000,000,007取模后输出即可)
输入:
多组数据每组数据一行,包含一个整数n,1<n<1,000,000,000
输出:
每组数据输出一个整数,即为f(n)对1,000,000,007取模后的结果
分析:
从题目中可以看出,数据范围比较大,如果是用递归或者递推方式写,肯定会超时,所以我们应该要考虑其他的方法。我们就可以考虑快速幂的方法。不过,这道题要用到矩阵乘法,用矩阵快速幂。
代码:
#include<iostream>
#define MOD 1000000007
using namespace std;
typedef long long ll;
struct matrix{
ll m[3][3];
}Mat,unit;
ll a[3]={0,1,2};
matrix multi(matrix a,matrix b){
matrix res;
ll x;
ll i,j,k;
for(i=0;i<3;i++){
for(j=0;j<3;j++){
x=0;
for(k=0;k<3;k++){
x=(x+((ll)a.m[i][k]*(ll)b.m[k][j]))%MOD;
}
res.m[i][j]=x%MOD;
}
}
return res;
}
void Init(){
Mat.m[0][0]=2;Mat.m[0][1]=1;Mat.m[0][2]=0;
Mat.m[1][0]=1;Mat.m[1][1]=0;Mat.m[1][2]=1;
Mat.m[2][0]=0;Mat.m[2][1]=0;Mat.m[2][2]=0;
for(int i=0;i<3;i++){
unit.m[i][i]=1;
}
}
matrix pow_matrix(matrix a,ll n){
matrix res=unit;
while(n){
if(n&1)res=multi(res,a);
a=multi(a,a);
n=n>>1;
}
return res;
}
ll Feibonaqi(ll n){
if(n==0)return a[0];
if(n==1)return a[1];
if(n==2)return a[2];
matrix res=pow_matrix(Mat,n-2);
return (a[2]*res.m[0][0]+a[1]*res.m[1][0])%MOD;
}
int main(){
Init();
ll n;
while(cin>>n){
cout<<Feibonaqi(n)<<endl;
}
}
a2 a1 a0=[ 2 , 1 , 0 ],Feibonaqi前三项
Mat:根据Feibonaqi推出的矩阵
2 1 0
1 0 0
0 0 0
a*Matn-2得到的一行三列的矩阵的第0项就是要求的值
如有错误欢迎指正!