题目描述
给定a0,a1,以及an=p*a(n-1) + q*a(n-2)中的p,q。这里n >= 2。 求第k个数对10000的模。
输入描述
输入包括5个整数:a0、a1、p、q、k。
输出描述
第k个数a(k)对10000的模。
示例1
20 1 1 14 5
输出
8359
代码
#include <stdio.h>
#include <stdlib.h>
void MetrixMul(int p1[2][2],int p2[2][2])
{
int t[2][2];
t[0][0]=(p1[0][0]*p2[0][0]%10000+p1[0][1]*p2[1][0]%10000)%10000;
t[0][1]=(p1[0][0]*p2[0][1]%10000+p1[0][1]*p2[1][1]%10000)%10000;
t[1][0]=(p1[1][0]*p2[0][0]%10000+p1[1][1]*p2[1][0]%10000)%10000;
t[1][1]=(p1[1][0]*p2[0][1]%10000+p1[1][1]*p2[1][1]%10000)%10000;
p1[0][0]=t[0][0];
p1[0][1]=t[0][1];
p1[1][0]=t[1][0];
p1[1][1]=t[1][1];
}
void MetrixPow(int p[2][2],int n)
{
int t[2][2];
t[0][0]=p[0][0];
t[0][1]=p[0][1];
t[1][0]=p[1][0];
t[1][1]=p[1][1];
if(n==1)
return;
else if(n%2)
{
MetrixPow(p,n-1);
MetrixMul(p,t);
}
else
{
MetrixPow(p,n/2);
MetrixMul(p,p);
}
}
int main()
{
int a0,a1,p,q,k;
int pa[2][2];
while(scanf("%d%d%d%d%d",&a0,&a1,&p,&q,&k)!=EOF)
{
if(k==0)
{
printf("%d\n",a0%10000);
continue;
}
if(k==1)
{
printf("%d\n",a1%10000);
continue;
}
pa[0][0]=p%10000;
pa[0][1]=q%10000;
pa[1][0]=1;
pa[1][1]=0;
MetrixPow(pa,k-1);
printf("%d\n",(pa[0][0]*a1+pa[0][1]*a0)%10000);
}
return 0;
}
解析
题目参考9度OJ的讲解,博客地址:https://blog.csdn.net/lhyer/article/details/47841763
本题看似简单,其实很容易超时超内存,其实本题是一道快速幂矩阵的题目。
通过对题目分析,我们可以得到矩阵地推公式如下:
![在这里插入图片描述](https://img-blog.csdnimg.cn/20190222095418962.jpg)
将公式右边推至a1, a0即可得:
![在这里插入图片描述](https://img-blog.csdnimg.cn/20190222095151297.png)
然后这个题即转化为快速幂矩阵问题。