uva 12169 不爽的裁判 例题2

思路很清晰就是枚举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
*/