我们可以暴力枚举a,然后通过x1和x3确定b的值,然后确定其他的数字,一旦出现错误就放弃这组解。
关键问题就在于如何通过a,x1,x3确定b的值
x2 = ( x1 * a + b) % M
x3 = ( x2 * a + b ) % M = ( ( x1 * a + b ) % M * a + b ) % M
x3 - k * M = x1 * a % M * a % M + b * a % M + b
我们不妨设b<M
b * ( a + 1 ) + k * M = x3 - x1 * a * a
我们想要求出b,即就是求出这个二元一次方程的一组解,这时候用扩展欧几里得算法就可以解决。(虽然他的原理我还没有搞懂)
可是写出来以后我却一直连样例都过不了,经过一个多小时的探索,我发现,爆longlong了,你永远不知道哪里会爆long long,无语。
得到的经验,凡是对空间复杂度要求不是很高又有乘法的运算还是尽量用long long ,可以避免爆long long 而引发的错误。
经过反复调试,我终于在没有用long long 的情况下做出了这道题(心酸)
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<climits>
#include<cctype>
#include<queue>
#include<set>using namespace std;typedef long long ll;
const int INF=0x3f3f3f3f;
const int MAXN=105;
const int MOD=10001;
int a1[MAXN],a2[MAXN];
int n;
int a,b;
int c,t,p,q,k;void gcd_ojbk(int a,int b,int& d,int& x,int& y)
{if(!b){d=a; x=1; y=0;}else{gcd_ojbk(b,a%b,d,y,x);y-=x*(a/b); }
}int main()
{scanf("%d",&n);for(ll i=1;i<=n;i++){scanf("%d",&a1[i]);}for(a=0;a<MOD;a++){bool flag=true;t=a+1;p=(a1[2]-a1[1]*a%MOD*a%MOD+MOD)%MOD;gcd_ojbk(MOD,t,q,c,k);if(p%q) continue;b=((k*(p/q)%MOD+MOD)%MOD); //这里很容易爆long long a2[1]=(a*a1[1]%MOD+b)%MOD;for(int i=2;i<=n;i++){if((a*a2[i-1]%MOD+b)%MOD!=a1[i]){flag=false;break;}a2[i]=(a*a1[i]%MOD+b)%MOD;}if(flag){for(int i=1;i<=n;i++){printf("%d\n",a2[i]);}break;}}return 0;
}