题目链接: Fibonacci Check-up
思路
我们应该已经会了基本的求斐波那契这种递推式很简单的矩阵快速幂了,问题是本题没让你求斐波那契,只不过求和的公式里用到一部分而已。我们可以看看S(n)和F(n)之间有没有什么联系,可以先打个表。
(如果只是用公式推,网上有详细过程,非人类的推导过程)
F1 | F2 | F3 | F4 | F5 | F6 |
---|---|---|---|---|---|
1 | 1 | 2 | 3 | 5 | 8 |
S1 | S2 | S3 | S4 | S5 | S6 |
---|---|---|---|---|---|
1 | 3 | 8 | 21 | 55 | 144 |
可以发现,S(n) = F(2 * n) ,因此这就转化成了求斐波那契的某一项了。
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <cstring>
using namespace std;
struct Matrix{
long long m[4][4];
};
long long n,m;
Matrix mul(Matrix A,Matrix B)
{
Matrix tmp;
memset(tmp.m,0,sizeof(tmp.m));
for(int i = 1; i <= 2; ++i)
{
for(int k = 1; k <= 2; ++k)
if(A.m[i][k])
for(int j = 1; j <= 2; ++j)
{
tmp.m[i][j] = (tmp.m[i][j] + A.m[i][k] * B.m[k][j]) % m;
}
}
return tmp;
}
Matrix pow(Matrix A,long long n)
{
Matrix res;
memset(res.m,0,sizeof(res));
for(int i = 1; i <= 2; ++i)
res.m[i][i] = 1;
while(n)
{
if(n & 1)
res = mul(res,A);
A = mul(A,A);
n >>= 1;
}
return res;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
cin >> n >> m;
if(n == 0)
{
cout << '0' << endl;
continue;
}
Matrix A,B;
A.m[1][1] = 1;
A.m[1][2] = 1;
A.m[2][1] = 1;
A.m[2][2] = 0;
A = pow(A,2 * n - 1);
B.m[1][1] = 1;
B.m[2][1] = 0;
B = mul(A,B);
cout << B.m[1][1] << endl;
}
}