[WIKIOI 1250]Fibonacci数列(数论+数学+矩阵乘法+二分快速幂)

题目描述 Description

定义：f0=f1=1, fn=fn-1+fn-2(n>=2)。{fi}称为Fibonacci数列。

输入n，求fn mod q。其中1<=q<=30000。

输入描述 Input Description

第一行一个数T(1<=T<=10000)。

以下T行，每行两个数，n，q(n<=109, 1<=q<=30000)

输出描述 Output Description

文件包含T行，每行对应一个答案。

样例输入 Sample Input

3

6 2

7 3

7 11

样例输出 Sample Output

1

0

10

数据范围及提示 Data Size & Hint

1<=T<=10000

n<=109, 1<=q<=30000

二分快速幂可以将时间降到log以2为的n, 矩阵需要自己构造, 空间递归不需太多, 其实重点就这些了.

#include <stdio.h>
#include <string.h>
#define MAXN 305
#define INF 100000000
#define eps 1e-7
#define PI 3.1415926535898
using namespace std;
int n = 2, m, q;
long long tt[2][2]={{1, 1}, {1, 0}};
struct wwj
{
    int r, c;
    long long mat[3][3];
} need, ready;
void init()
{
    memset(need.mat, 0, sizeof(need.mat));
    need.r = n;
    need.c = n;
    for(int i = 1; i <= n; i++)
    {
        need.mat[i][i] = 1;
    }
    ready.c = n;
    ready.r = n;
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= n; j++)
        ready.mat[i][j] = tt[i - 1][j - 1];
    }
}
wwj multi(wwj x, wwj y)
{
    wwj t;
    int i, j, k;
    memset(t.mat, 0, sizeof(t.mat));
    t.r = x.r;
    t.c = y.c;
    for(i = 1; i <= x.r; i++)
    {
        for(k = 1; k <= x.c; k++)
            if(x.mat[i][k])
            {
                for(j = 1; j <= y.c; j++)
                {
                    t.mat[i][j] += (x.mat[i][k] * y.mat[k][j]) % q;
                    t.mat[i][j] %= q;
                }
            }
    }
    return t;
}
int main()
{
    int m;
    int CASES;
    scanf("%d", &CASES);
    while(CASES--)
    {
        scanf("%d %d", &m, &q);
        ++m;
        if(m == -1) break;
        init();
        while(m)
        {
            if(m & 1)
            {
                need = multi(ready, need);
            }
            ready = multi(ready, ready);
            m = m >> 1;
        }
        printf("%lld\n", need.mat[1][2]);
    }
    return 0;
}