首先本题直接暴力枚举a,b可解,最坏为1秒左右时间可解出(一个test文件需要的时间)
高效算法为枚举a,然后由x1,x3,解出b
a*x1+b - MOD*y1 = x2;
a*x2+b - MOD*y2 = x3;
解得:
x3 - a*a*x1=(a+1)*b + MOD * y;
该方程为关于变量b的模线性方程 ,用扩展欧几里得算法解出一个解b0,(当gcd(a+1,MOD)==1) 则解出的为一个同余系;
b = b0 + MOD*k (k为任意整数);(该方程对应了 b = b0 + MOD' * k ,其中MOD' 为MOD/ gcd(a+1,MOD) ); 只需要检验一个b即可
但是当gcd(a+1,MOD)不等1时,直接用b0求解是有问题的因为解不在是MOD的同余系而是MOD‘的同余系;
所以正解应该是算出b0然后解出0 - 10000范围内的 可行b 然后检验; 算法复杂度为 O(nlogn*100);
本人较懒只写出了认为gcd(a+1,MOD)==1在所有时刻成立的版本 ;
#include <cstdio>
#include <cmath>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
void gcd(LL a,LL b,LL& d,LL& x,LL& y){
if(!b){d=a; x=1; y=0;}
else {gcd(b,a%b,d,y,x); y-=x*(a/b);}
}
const int MOD = 10001;
const int maxn = 101;
LL ans[maxn*3],te[maxn*3];
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=2*n;i+=2){
scanf("%lld",&ans[i]);
}
int lim = 2*n;
LL x,y,g;
bool flag =false;
for(int a=0;a<MOD;a++){
gcd(a+1,MOD,g,x,y);
LL c =ans[3] - ans[1]*a*a;
if(c%g!=0) continue;
LL b=x*c/g;
te[1]=ans[1];
int ok=1;
for(int i=2;i<=lim;i++){
te[i]=(te[i-1]*a+b)%MOD;
if((i&1) && te[i] != ans[i]){
ok=false; break;
}
}
if(ok){ flag=true; break; }
}
for(int i=2;i<=lim;i+=2){
printf("%d\n",te[i]);
}
return 0;
}