【Tenka1 Programmer Contest 2017 F】ModularPowerEquation!!（欧拉定理）

题目大意

输入A,M,找任意一个K ,使得$A^K≡K(mod\space M)$

题解

欧拉定理

若n,a互质，则

$$a^{\phi(n)}≡1(mod\space n)$$

证明：
令1~n中，与n互质的数为$x_1,x_2,x_3,...,x_{\phi(n)}$
令$m_i=ax_i$,

1. 性质1：所有$m_i$中，任意两个都不关于n同余。
   若存在 $m_i≡m_j(mod\space n)$
   则$n|m_i-m_j$，$n|a(x_i-x_j)$，∵n与a互质，$x_i-x_j<n$，$n$不整除$a(x_i-x_j)$，矛盾。
2. 性质2：所有$m_i\space mod \space n$与n互质。
   令$m_i=pn+r$，若r与n有公因子q，则$m_i$也一定有因子q，$m_i$与$n$不互质
   ∵a与n互质，x_i与n互质，$ax_i$不含n的任何因子，矛盾。

由以上性质得出
$m_1,m_2,m_3,...,m_{\phi(n)}$所有数模n，得到的数一定是$x_1,x_2,x_3,...,x_{\phi(n)}$
∴$m_1m_2m_3...m_{\phi(n)}≡x_1x_2x_3...x_{\phi(n)}(mod\space n)$
$(a^{\phi(n)}-1)x_1x_2x_3..x_{\phi(n)}≡0(mod\space n)$
$n|(a^{\phi(n)}-1)x_1x_2x_3..x_{\phi(n)}$
$n|a^{\phi(n)}-1$
$a^{\phi(n)}≡1(mod\space n)$

缩小问题规模

$A^K≡K(mod\space M)$
$A^{\phi(M)}...A^{\phi(M)}A^{K\space mod\space \phi(M)}≡K(mod\space M)$
∵$A^{\phi(M)}≡1(mod\space M)$
∴$A^{K\space mod\space \phi(M)}≡K(mod\space M)$

找一个t，使得$t≡K(mod\space \phi(M))$
∴$A^t≡A^K≡K(mod\space M)$
$A^t\space mod\space M=t\space mod\space \phi(M)$
∴$A^t≡t(mod\space gcd(M,\phi(M)))$
然后问题变为求t，M缩小。

具体实现

当M=1时，K=1。

用$O(\sqrt M)$暴力求出$\phi(M)$，取$gcd(M,\phi(M))$递归求解，得到一个t.
根据以下条件
$A^t≡K(mod\space M)$
$A^t\space mod\space M=t\space mod\space \phi(M)=K$
得出
$x×M+A^t\space mod\space M=y×\phi(M)+t\space mod\space \phi(M)$
$x×M-y×\phi(M)=t\space mod\space \phi(M)-A^t\space mod\space M$
可用扩展欧几里得解出这个不定方程的x.
为保证得到正数，适当的把t扩大，加上几个$\phi(M)$使等式右边变为正数.
解出x后
$K=x×M+A^t\space mod\space M$

代码

#include<cstdio>
long long modpow(long long a,long long b,long long p)
{
    long long ret=1;
    for(;b>0;b>>=1,a=(a*a)%p)
        if(b&1)
            ret=(ret*a)%p;
    return ret;
}
long long phi(long long n)
{
    long long res=n,t=n;
    for(long long i=2;i*i<=t;i++)
        if(t%i==0)
        {
            res-=res/i;
            while(t%i==0)
                t/=i;
        }
    if(t>1)
        res-=res/t;
    return res;
}
long long gcd(long long a,long long b)
{
    if(b==0)
        return a;
    return gcd(b,a%b);
}
long long exgcd(long long a,long long b,long long &x,long long &y)
{
    if(b==0)
    {x=1;y=0;return a;}
    long long g;
    g=exgcd(b,a%b,y,x);
    y-=a/b*x;
    return g;
}
long long solve(long long A,long long M)
{
    if(M==1)
        return 1;
    long long pM,At,d,t,x,y;
    pM=phi(M);
    d=gcd(M,pM);
    t=solve(A,d);
    At=modpow(A,t,M);
    exgcd(M/d,pM/d,x,y);
    x=x*(((t-At)%pM+pM)%pM)/d;
    x=((x%(pM/d))+pM/d)%(pM/d)+pM/d;
    return x*M+At;
}
int main()
{
    int Q;
    long long A,M;
    scanf("%d",&Q);
    while(Q--)
    {
        scanf("%lld%lld",&A,&M);
        printf("%lld\n",solve(A,M));
    }
    return 0;
}