Question

杭电1005 为什么会WA

我的代码
#include "iostream"
using namespace std;
int main()
{
int a,b,num[8],i;
long n;
num[1]=num[2]=1;
while(scanf("%d %d %ld",&a,&b,&n),a+b+n)
{
for(i=3;i<=6;i++)
num[i%6]=(a*num[i-1]+b*num[i-2])%7;
printf("%d\n",num[n%6]);
}
return 0;
}
是个循环，这应该没错吧~
高人指点一下，先谢谢了~
http://acm.hdu.edu.cn/showproblem.php?pid=1005
最好能把我答案错的地方指出来~谢谢了~

Answer 1

循环周期是49，而不是7，修改如下：

//---------------------------------------------------------------------------
#include <stdio.h>

int main(void)
{
int a,b,i;
long int f[49],n;

while (scanf("%d%d%ld",&a,&b,&n)!=EOF)
{
if (!(a+b+n)) break;

f[0]=f[1]=1;
for (i=2; i<49; i++) {
f[i]=(a*f[i-1]+b*f[i-2])%7;
}
printf("%d\n",f[(n)%48-1]);
}

}
//---------------------------------------------------------------------------