poj上的题,比赛结束之后学长跟我说这个题在poj上做出来的只有200多个人,,,,顿时有种不想做了的感觉,晚上回来研究了一会发现自己的思路应该是正确的,就是矩阵乘法加快速幂,时间上肯定没问题,就是不知道错在哪,把代码给学长看了一下,学长说要考虑n<k的情况,我一看自己确实漏了这个,加上之后还是WA。。。。这就蛋疼了,然后就仔仔细细看了好几遍代码,终于发现是在写矩阵乘法的函数时少了一步模除。。。。。
#include<stdio.h>
#include<string.h>
int x,k,a[110],b[110],c[110][110],e[110][110];
void f(int c[110][110],int d[110][110])
{
int dd[110][110];
for(int i=1;i<=k;i++)
for(int j=1;j<=k;j++)
{
int t=0;
for(int l=1;l<=k;l++)
{
t+=(c[i][l]*d[l][j])%10000;
}
dd[i][j]=t%10000;\\就是在这个地方,一开始漏掉了这个模除。。
}
for(int i=1;i<=k;i++)
for(int j=1;j<=k;j++)
c[i][j]=dd[i][j];
}
void mi(int n)
{
while(n!=0)
{
if(n%2==1)
f(e,c);
f(c,c);
n/=2;
}
}
int main()
{
int sum=0;
while(scanf("%d",&k)!=EOF)
{
sum=0;
if(k==0) break;
memset(c,0,sizeof(c));
memset(e,0,sizeof(e));
for(int i=1;i<=k;i++)
for(int j=1;j<=k;j++)
if(i==j) e[i][j]=1;
for(int i=0;i<k;i++)
scanf("%d",&a[i]);
for(int i=1;i<=k;i++)
{
scanf("%d",&b[i]);
c[1][i]=b[i];
}
for(int i=2;i<=k;i++)
{
for(int j=1;j<=k;j++)
{
if(i-1==j) c[i][j]=1;
}
}
scanf("%d",&x);
if(x>=k)
{
mi(x-k+1);
for(int i=1;i<=k;i++)
sum+=(e[1][i]*a[k-i])%10000;
sum=sum%10000;
printf("%d\n",sum);
}
else
printf("%d\n",a[x]);
}
return 0;
}
加上之后果然就过了,,,阿门!