uva 12169 Disgruntled Judge

法一直接暴力枚举a和b的值，法二扩展欧几里德算法。

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
const int mod=10001;

int main()
{
    int x[205],t,i,flag,a,b,j;
    while(~scanf("%d",&t))
    {
        for(i=0;i<t;i++)
        {
            scanf("%d",&x[2*i+1]);
        }

        for(a = 0; a <= 10000; a++)
        {
            for(b = 0; b <= 10000; b++)
            {
                flag=1;
                x[2]=(a*x[1]+b)%mod;
                for(i=3;i<=2*t;i++)
                {
                    if(i%2==0)
                    {
                        x[i]=(x[i-1]*a+b)%mod;
                    }
                    else
                    {
                        if(x[i]!=(a*x[i-1]+b)%mod)
                        {
                            flag=0;
                            break;
                        }
                    }
                }
                if(flag) break;
            }
            if(flag) break;
        }
        for(i = 1; i <= t; i++)
        {
            printf("%d\n",x[2*i]%10001);
        }
    }
    return 0;
}

法二
gcd里面没有用longlong挂了好久……

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
const int mod=10001;
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-=(a/b)*x;
    }
}

int main()
{
    int x[205],t,i,flag;
    ll k,a,b,p,d;
    while(~scanf("%d",&t))
    {
        for(i=0; i<t; i++)
        {
            scanf("%d",&x[2*i+1]);
        }

        for(a=0; a<=10000; a++)
        {
            p=x[3]-x[1]*a*a;
            gcd(mod,a+1,d,k,b);
            if(p%d!=0) continue;
            b*=p/d;
            flag=1;
            for(i=2; i<=2*t; i++)
            {
                if(i%2==0)
                {
                    x[i]=(x[i-1]*a+b)%mod;
                }
                else
                {
                    if(x[i]!=(a*x[i-1]+b)%mod)
                    {
                        flag=0;
                        break;
                    }
                }
            }
            if(flag) break;
        }

        //printf("%d %d\n",a,b);
        for(i=1; i<=t; i++)
        {
            printf("%d\n",x[2*i]);
        }
    }
    return 0;
}