题目链接:点击打开链接
题目描述:
给出三个数x1,a,b,然后根据递推公式xi=(axi-1+b)mod10001,计算出了一个长度为2T的数列。然后把T和x1,x3,x5··· x2T-1写到输入文件,x2,x4,x4,···x2T作为输出文件
输入保证T<=100,如果有多种可能,任意输出一种即可
解题思路:如果知道a,我们就可以通过x1,x3,计算出b。有了x1,a,b我们就可以在O(T)的时间内求的整个序列。如果在计算中发现和输入矛盾,则这个a是非法的。由于mod10001,所以a是0~10000的整数,所以我们可以考虑枚举a,时间最差也是10000*200.根据x3=(a^2*x1+ab+b)mod 10001所以(a+1)b+k*10001=x3-a^2*x1。所以我们可以利用扩展欧几里德算法求解b。注意a^2*x1的范围可能会超过int,所以long long即可
代码如下:
#include <cstdio>
#define MOD 10001
using namespace std;
int d[210];
int n;
void gcd(int a,int b,int& d,int &x,int &y){
if(!b){d=a;x=1;y=0;}
else{gcd(b,a%b,d,y,x);y-=x*(a/b);}
}
int main(){
while(scanf("%d",&n)==1){
for(int i=0;i<n;i++)
scanf("%d",&d[(i<<1)+1]);
//n==0,no
d[2]=d[1];///n==1,a=1,b=0
//n>=2
long long a,b;
if(n>=2){
for(a=0;a<10001;a++){
int gd,x,y;
gcd(a+1,MOD,gd,x,y);
if((d[3]-a*a*d[1])%gd==0){///a*a*d[1]可能溢出long long
b=x*((d[3]-a*a*d[1])/gd);
int i;
for(i=1;i<n;i++){
d[i<<1]=(d[(i<<1)-1]*a+b)%MOD;
if(d[(i<<1)+1]!=((d[(i<<1)]*a+b)%MOD))
break;
}
d[n<<1]=(d[(n<<1)-1]*a+b)%MOD;
if(i==n)
break;
}
}
}
for(int i=1;i<=n;i++){
printf("%d\n",d[i<<1]);
}
}
return 0;
}