矩阵快速幂与快速幂(费马小定理的应用)
M斐波那契数列F[n]是一种整数数列,它的定义如下:
F[0] = a
F[1] = b
F[n] = F[n-1] * F[n-2] ( n > 1 )
现在给出a, b, n,你能求出F[n]的值吗?
Input
输入包含多组测试数据;
每组数据占一行,包含3个整数a, b, n( 0 <= a, b, n <= 10^9 )
Output
对每组测试数据请输出一个整数F[n],由于F[n]可能很大,你只需输出F[n]对1000000007取模后的值即可,每组数据输出一行。
Sample Input
0 1 0
6 10 2
Sample Output
0
60
对于这道题来说,是一个简单的矩阵快速幂与快速幂的模板题,但是难就难在对费马小定理的应用上。费马小定理(https://baike.baidu.com/item/费马小定理/4776158?fr=aladdin);
大家应该对1e9+7这个数字非常敏感才对,在一些题目中极有可能会用到费马小定理。
通过找规律发现:
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long ll;
struct mat{
long long a[5][5];
};
const long long mod=1000000007;
mat Mulity(mat a,mat b){//矩阵快速幂乘法,也可以重载*运算符 mat operater*(mat a,mat b)函数部分相应变化
mat ans;
memset(ans.a,0,sizeof(ans.a));
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
for(int k=0;k<2;k++)
ans.a[i][j]+=a.a[i][k]*b.a[k][j]%(mod-1);
ans.a[i][j]%=(mod-1);
}
}
return ans;
}
mat Power(ll k){//求一个矩阵的k次方
mat x;//x为斐波那契数列的单位矩阵
mat t;//t为单位矩阵
t.a[0][0]=1;t.a[0][1]=0;
t.a[1][0]=0;t.a[1][1]=1;
x.a[0][0]=1;x.a[0][1]=1;
x.a[1][0]=1;x.a[1][1]=0;
while(k){
if(k&1)
t=Mulity(t,x);
k>>=1;
x=Mulity(x,x);
}
return t;
}
long long qpower(long long a,long long b){
long long s=1;
while(b){
if(b&1) {
s=s%mod;
a=a%mod;
s=s*a; //先模在乘,不然容易超时间
}
a=a%mod;
a=a*a;
b>>=1;
}
return s%mod;
}
int main(){
long long a,b,n,sum1,sum2;
while(~scanf("%lld%lld%lld",&a,&b,&n)){
if(n==0) printf("%lld\n",a);
else if(n==1) printf("%lld\n",b);
else{
mat x=Power(n);
ll be=x.a[0][1];
ll ae=x.a[1][1];
sum1=qpower(a,ae);
sum2=qpower(b,be);
sum1=(sum1*sum2)%mod;
printf("%lld\n",sum1);
}
}
return 0;
}
通过这道题,了解了费马小定理,其实也就是一个公式,慢慢加油努力,追赶上别人。