UVA - 12169 Disgruntled Judge (模线性方程的解为同余系)

首先本题直接暴力枚举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;
}



你可能感兴趣的:(UVA - 12169 Disgruntled Judge (模线性方程的解为同余系))