Codechef August Challenge 2018 : Modular GCD

传送门

一开始还手动拓欧找规律，发现好像玩不了。

然后想了想，A-B这个数比较小，枚举它的因子判断合不合法就行了。

需要特判A=B的情况。

#include<cstdio>
#include<algorithm>
#define ll long long
#define ld long double
using namespace std;

ll a,b,n,c,t,d,A,B;
int i;
const int MOD=1e9+7;
inline void MM(ll &a,ll M){while(a>=M)a-=M;}
inline ll cc(ll x,ll y,ll M){
    x=x*y-(ll)(((ld)x*y+0.01)/M)*M;
    return x<0?x+M:x;
}
inline ll mi(ll x,ll y,ll M){
    ll ans=1;
    x%=M;
    while(y){
        if (y&1) ans=cc(ans,x,M);
        y>>=1;x=cc(x,x,M);
    }
    return ans;
}
inline bool ju(ll x){
    return (mi(a,n,x)+mi(b,n,x))%x==0;
}
inline void work(){
    scanf("%lld%lld%lld",&a,&b,&n);
    c=a-b;
    if (c==0){
        printf("%d\n",(mi(a,n,MOD)+mi(b,n,MOD))%MOD);
        return;
    }
    for (i=1;1LL*i*i<=c;i++)
    if (c%i==0&&ju(c/i)){
        printf("%lld\n",c/i%MOD);
        return;
    }
    for (i--;i;i--)
    if (c%i==0&&ju(i)){
        printf("%lld\n",i%MOD);
        return;
    }
}
int main(){
    scanf("%lld",&t);
    while(t--) work();
}

View Code