UVA 11582 巨大的斐波那契額數列！ 快速冪取模

從現在開始做紫皮書上的題， 本題是一道斐波那契數列的題， 結合快速冪， 水題。

輸入a, b, n, 計算f( a ^ b) % n  ， f( i ) = f( i - 1) + f( i - 2 ) ， f( 0 ) = f(1) = 1, 其中，0 <= a , b <= 2 ^ 64,  0 <= n <= 1000

以后少廢話， 直接存代碼

// 計算 f(a^b) mod n, 其中 f(i) = f(i-1) + f(i-2), f(0) = f(1) = 1, 其中0 <= a, b < 2^64 , n <= 1000
#include <cstdio>
#include <iostream>
#include <cstring>
#include <set>
using namespace std;

#define MAXN 1007
typedef unsigned long long LL;
int f[MAXN*MAXN];

int fun( LL a, LL b, int n ) {
    LL ans = 1;
    a = a % n;
    while( b > 0 ) {
        if( b & 1 ) ans = ans * a % n;
        b /= 2;
        a = a * a % n;
        
    }
    return (int)ans;
}
int main() {
    LL a, b;
    int n;
    int t;
    cin >> t;
    while( t-- ) {
        cin >> a >> b >> n;
        memset( f, 0, sizeof(f) );
        f[0] = f[1] = 1;
        int i = 0;
        
        for( i = 2; i <= n * n; i++ ) {
            f[i] = ( f[i-1] % n + f[i-2] % n ) % n;
        }
        
        cout << f[fun( a, b, n )-1] << endl;
    }
    
}