数学序列
【问题描述】
已知一个函数f:
f(1)=1
f(2)=1
f(n)=(a×f(n-1)+b×f(n-2))mod7
现给出a,b,n,要你求出f(n).
【输入文件】
每一行输入一组数据分别为A,B,N(1<=A,B<=1000,1<=N<=200000000)
【输出文件】
每一行输出结果f(n).
【输入样例】
113
1210
【输出样例】
2
5
略微超前,这个是矩阵乘法,朴素过5组。(但是当时手贱,把循环变量打成n了,过了两组)。有一个教训,我用的递推,但是本来用两个变量就可以递推的,我开了个f数组。
矩阵乘法。基本算法:
因此
这既是一个菲波拉契数列的求法
(任何递推公式都可以用这个)
即可求得菲波拉契数列第n项。即p+q。
既然是求幂,就用O(lgn)的快速幂可以快速得到答案。(注意同余定理)
另外最好将矩阵的乘法单独写一个模块,这样代码精简得多,不易出错
另外
另外一个思路。这个数列有周期性。(注意是取余了的,菲波拉契数列当然没有周期)
比赛第三题就用了这个方法。只是人品太差,只得了10分。
#include <iostream>
long a[3][3];
long c[3][3];
long b[3][3];
long t[3][3];
void multi(long p[3][3],long q[3][3],long n)
{
for (long i=1;i<3;i++)
for (long j=1;j<3;j++)
t[i][j]=(p[i][1]*q[1][j]%n+p[i][2]*q[2][j]%n)%n;
for (long i=1;i<3;i++)
for (long j=1;j<3;j++)
p[i][j]=t[i][j];
}
void set(long p[3][3],long q[3][3])
{
for (long i=1;i<3;i++)
for (long j=1;j<3;j++)
p[i][j]=q[i][j];
}
long kysumi(long m,long n)
{
set(b,a);
while (m>0)
{
if (m&1==1) multi(c,a,n);
multi(a,a,n);
m>>=1;
}
}
int main()
{
freopen("number.in","r",stdin);
freopen("number.out","w",stdout);
long n;long p;long q;
std::cin >> p >> q >> n;
while(std::cin)
{
a[1][1]=0;
a[1][2]=1;
a[2][1]=q;
a[2][2]=p;
c[1][1]=1;
c[1][2]=0;
c[2][1]=0;
c[2][2]=1;
kysumi(n-2,7);
std::cout << (c[2][1]+c[2][2])%7 << std::endl;
std::cin >> p >> q >> n;
}
}