2018年东北农业大学春季校赛 K-wyh的数列
链接:https://www.nowcoder.com/acm/contest/93/K
来源:牛客网
题目描述
wyh学长特别喜欢斐波那契数列,F(0)=0,F(1)=1,F(n)=F(n-1)+F(n-2)(n>=2)
一天他突发奇想,想求F(a^b)%c
输入描述:
输入第一行一个整数T(1<=T<=100),代表测试组数
接下来T行,每行三个数 a,b,c (a,b<=2^64) (1
输出描述:
输出第a^b项斐波那契数对c取余的结果
示例1
输入
3
1 1 2
2 3 1000
32122142412412142 124124124412124 123
输出
1
21
3
思路
大数取模,幂取模,模的计算方法
已知:
(a+b)%n=(a%n+b%n)%n
(a-b)%n=(a%n-b%n+n)%n
(a*b)%n=(a%n*b%n)%n
显然:
F[i] % n = (F[i-1]%n + F[i-2]%n) % n
寻找循环节替代mod。发现因为fi是由前两项确定的,只要前两项和f[0],f[1]一样,就找到了循环节。
AC代码
#include <iostream>
#include <string>
#include <cmath>
using namespace std;
typedef unsigned long long ull;
const int maxn=1000+10;
ull F[maxn*maxn];
// powerMod(a, b, mod)相当于 a^b%mod
int powerMod(ull a, ull b, int mod) {
ull result = 1;
while(b) {
if(b&1) {
result = (a*result)%mod;
}
a = (a*a)%mod;
b/=2;
}
return result;
}
int main() {
int T;
cin >> T;
while(T--) {
ull a,b,c,mod;
int flag=1;
cin >> a >> b >> c;
F[0]=0;
F[1]=1;
for(ull i=2; i<=c*c; i++) {
F[i]=(F[i-1]%c+F[i-2]%c)%c;
if(F[i]==F[1]&&F[i-1]==F[0]) {
mod=i-1;
break;
}
}
if(a == 0 || c == 1)
cout << 0 << endl;
else
cout<<F[powerMod(a%mod, b, mod)]<<endl;
}
return 0;
}