2018年东北农业大学春季校赛 K-wyh的数列

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;
}