BZOJ2976 出圈游戏 [EX_CRT]

2976: [Poi2002]出圈游戏

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 110   Solved: 30
[ Submit][ Status][ Discuss]

Description

思考

扩展GCD解决mod非质数的扩展CRT问题。
第i轮有n−i+1个人，记为mi，n只有20，暴力算出该轮出圈的是第几个，记为ri。
$->$
$Ans≡r_1 mod(m_1)$
$Ans≡r_2 mod(m_2)$
$⋮ ⋮ ⋮$
$Ans≡r_{n-1} mod(m_{n−1})$
最后一个没有用了，只用n−1个方程就行了。然后就是$ri可能=mi$，所以每一个ri要对mi求余。
然后用ExCRT合并。还有一点就是$Ans!=0$

#include<bits/stdc++.h>
using namespace std;
int gcd(int a,int b){
    return b==0?a:gcd(b,a%b);
}
void ex_gcd(int a,int b,int &x,int &y){
    if(b==0){x=1;y=0;return ;}
    ex_gcd(b,a%b,x,y);
    int t=x;x=y;y=t-a/b*x;
}
int n,a[30],pos[30],d,r[30],M,ans,x,y,m[30];
int main(){
    scanf("%d",&n);
    for(register int i=1;i<=n;i++)
        scanf("%d",&a[i]),pos[a[i]]=i;
    for(register int i=1;i<n;i++){
        int k=pos[i],j=pos[i-1];
        while(j!=k){
            ++j;if(j>n)j=1;
            if(a[j]>=i)r[i]++;
        }
        m[i]=n-i+1;r[i]%=m[i];
    }
    M=m[1],ans=r[1];
    for(register int i=2;i<n;i++){
        ex_gcd(M,m[i],x,y);
        d=gcd(M,m[i]);
        if((r[i]-ans)%d){ans=-1;break;}
        ans=(r[i]-ans)/d*x%(m[i]/d)*M+ans;
        M*=m[i]/d;
        ans%=M;
    }
    if(ans!=-1)ans=(ans+M)%M;
    if(!ans)ans=M;
    if(ans==-1)puts("NIE");
    else printf("%d\n",ans);
    return 0;
}