Number Sequence
Total Submission(s): 178891 Accepted Submission(s): 44454
f(1) = 1, f(2) = 1, f(n) = (A * f(n - 1) + B * f(n - 2)) mod 7.
Given A, B, and n, you are to calculate the value of f(n).
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<stdlib.h>
#include<time.h>
#include<string>
#include<math.h>
#include<map>
#include<queue>
#include<stack>
#define INF 0x3f3f3f3f
#define ll long long
#define f(i,a,b) for(int i=a;i<b;i++)
#define sf(a) scanf("%d",&a)
#define sff(a,b) scanf("%d%d",&a,&b)
#define sfff(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define pf(a) printf("%d\n",a)
#define p printf("\n")
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
#define mod 1000000
int x[10000];
int main()
{
int a,b,n,c=1;
while(~sfff(a,b,n)&&(a,b,n))
{
if(a%7==0&&b%7==0)
{
if(n>2)pf(0);
if(n<=2)pf(1);
continue;
}
x[1]=1,x[2]=1;
f(i,3,10000)
{
x[i]=(a*x[i-1]+x[i-2]*b)%7;
if(x[i]==1)
{
if(x[i-1]==1)
{
c=i-2;
break;
}
}
}
int dd=n%c;
if(dd)pf(x[dd]);
else pf(x[c]);
}
}