51nod 1126 求递推序列的第N项

#include <bits/stdc++.h>
using namespace std;

map<int,map<int,int> > mp;
int f[1010];

int main()
{
int a,b,n,period,st,i;
while(cin>>a>>b>>n)
{
mp.clear();
f[1]=f[2]=1;
mp[1][1]=2;
for(i=3;i<=1000;i++)
{
f[i]=((a*f[i-1]+b*f[i-2])%7+7)%7;
if(mp[f[i-1]][f[i]])
{
period=i-mp[f[i-1]][f[i]];
st=mp[f[i-1]][f[i]];
break;
}
mp[f[i-1]][f[i]]=i;
}
if(n<st)
printf("%d\n",f
);
else
{
printf("%d\n",f[st+(n-st)%period]);
}
}
}