UVa12169（拓展欧几里得）

例题10-2　不爽的裁判（Disgruntled Judge, NWERC 2008, UVa12169）
有个裁判出的题太难，总是没人做，所以他很不爽。有一次他终于忍不住了，心 想：“反正我的题没人做，我干嘛要费那么多心思出题？不如就输入一个随机数，输出一个 随机数吧。”
于是他找了3个整数x1、a和b，然后按照递推公式xi=(axi-1+b) mod 10001计算出了一个长 度为2T的数列，其中T是测试数据的组数。然后，他把T和x1, x3,…, x2T-1写到输入文件中，x2, x4,…, x2T写到了输出文件中。
你的任务就是解决这个疯狂的题目：输入T, x1, x3,…, x2T-1，输出x2, x4,…, x2T。输入保 证T≤100，且输入的所有x值为0～10000的整数。如果有多种可能的输出，任意输出一个即 可。
【分析】
如果知道了a，就可以计算出x2，进而根据x3=(ax2+b) mod 10001算出b。有了x1、a和b， 就可以在O(T)时间内计算出整个序列了。如果在计算过程中发现和输入矛盾，则这个a是非 法的。由于a是0～10000的整数（因为递推公式对10001取模），即使枚举所有的a，时间效 率也足够高。

以上摘自刘汝佳前辈紫书。

不妨令mod=10001

有递推公式可得x3 = (ax2+b) %mod = (a (ax1 +b)%mod + b ) %mod;

所以x3 = [ a*a*x1 + (a+1)*b ] %mod;

即：b*(a+1) - k*mod = x3 - a*a*x1;  （k是整数）

由此可以用拓展欧几里得算法解得b。

得到此时a，b，带入递推式检查是否符合奇数项序列。

【代码】：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int mod=10001;
int x[11000],r[11100],T;
ll exgcd(ll a,ll b,ll &x,ll &y)
{
    if(!b){
        x=1;y=0;return a;
    }
    ll d=exgcd(b,a%b,y,x);
    y-=a/b*x;
    return d;
}
int main()
{
    cin>>T;
    for(int i=1;i<=2*T;i+=2)
        scanf("%lld",x+i);
    for(ll a=0;a<mod;a++)//枚举a!!注意用long long
    {
        ll c=(x[3]-a*a*x[1]);
        ll b,y;
        ll d=exgcd(a+1,mod,b,y);
        if(c%d)continue;
        b*=c/d;
        r[1] = x[1];

        int flag=1;//检查此时的a,b
        for(int i=2;i<=2*T;i++)
        {
            r[i]=(a*r[i-1]+b)%mod;
            if(i%2==1 && r[i]!=x[i]){
                flag=0;break;
            }
        }
        if(flag)
        {
            break;
        }
    }
    for(int i=2;i<=2*T;i+=2)
    {
        cout<<r[i]<<endl;
    }
}