思路很清晰就是枚举a求b然后Oncheck关键在于如何较快的求出b因为是在%10001的剩余系下
即 b*(a+1)=x3-a*a*x1+k*Mod 求解
移项 b*(a+1)+k*Mod=x3-a*a*x1 像什么?一元二次方程嘛,扩展欧几里得求解
#include<cstdio>
#include<cstring>
#include<iostream>
#define Mod 10001
#define LL long long
using namespace std;
LL T,x[330],ans[330];
void exgcd(LL a,LL b,LL& g,LL& x,LL& y){
if(b==0)x=1,y=0,g=a;
else{
exgcd(b,a%b,g,y,x);y-=x*(a/b);
}
}
// b*(a+1)=x3-a*a*x1+k*Mod
//b*(a+1)+k*Mod=x3-a*a*x1 求解
LL getb(LL a){
LL p=(x[3]-a*a*x[1]),q=a+1;
LL g,x,y;
exgcd(a+1,Mod,g,x,y);
if(p%g)return -1e9;
x=x*(p/g);
return x;
}
bool check(LL a,LL b){
for(LL i=2;i<=T*2;i+=2){
x[i]=(a*x[i-1]%Mod+b)%Mod;
if(x[i+1]!=(a*x[i]%Mod+b)%Mod&&i<T*2)return false;
}
return true;
}
int main(){
scanf("%lld",&T);
for(LL i=1;i<=T;i++)
scanf("%lld",x+(i-1)*2+1);
for(LL i=0;i<=10000;i++){
LL b=getb(i);
if(b==-1e9)continue;
if(check(i,b)){
for(LL j=2;j<=T*2;j+=2)printf("%lld\n",x[j]);
return 0;
}
}
return 0;
}/*
3
17
822
3014
*/