bzoj2219 数论之神

首先虽然题上没有说，但是可以当成保证有解来做，省掉很多麻烦的判断。
不妨记$2k+1=P$，$P=\prod p_i^{d_i}$，$p_i$是质数。对每一个$p_i^{d_i}$单独求解$x^A\equiv B \pmod {p_i^{d_i}}$的解的个数，因为每个方程得到的解都形如$x\equiv w\pmod {p_i^{d_i}}$，从每个方程中任选一个解都能通过中国剩余定理求出模$P$下的通解，因此只需要把这些解的个数乘起来就行了。
接下来考虑如何求解$x^A\equiv B\pmod {p^d}$。
首先，如果$B=0$，也就是$x^A$是$p^d$的倍数，这样$x$一定是$p^k$的倍数，其中$k$满足$Ak\ge d$，即$k=\lceil\frac dA \rceil$。解的个数是$\frac{p^d}{p^k}$。
否则的话，$B$可以写成$p^aq$，其中$a\ge 0$。记$x=p^ky$，其中$Ak=a$。约去$p^a$得$y^A\equiv q\pmod{p^{d-a}}$。对原根取对数得到$A\log y\equiv \log q\pmod{\varphi(p^{d-a})}$，这是一个二元一次不定方程，记$b=\varphi(p^{d-a})$，相邻两个解差$\frac{b}{\gcd(A,b)}$，因此解的个数是$\gcd(A,b)$。一个$\log y$对应一个$x$，但是注意求出的解$y$的范围是$[0,p^{d-a})$，而实际上$y$可以取到$[0,p^{d-k})$，再把答案扩大对应倍数就可以了。

#include<cstdio>
#include<cmath>
#include<map>
#include<vector>
#include<algorithm>
using namespace std;
int A,b;
int pow(int base,int k)
{
    int ret=1;
    for (;k;k>>=1,base*=base)
        if (k&1) ret*=base;
    return ret;
}
int gcd(int a,int b)
{
    return b?gcd(b,a%b):a;
}
int solve(int p,int d,int pd)
{
    int B=b%pd,f=1,a=0,k,phi,rt,ind,g;
    if (B==0) return pow(p,d-(d+A-1)/A);
    while (B%p==0)
    {
          B/=p;
          d--;
          pd/=p;
          a++;
    }
    k=a/A;
    phi=pd/p*(p-1);
    g=gcd(A,phi);
    return g*pow(p,a-k);
}
void solve()
{
     int k,p,m,d,ans=1,pd;
     scanf("%d%d%d",&A,&b,&k);
     p=2*k+1;
     m=sqrt(p+0.5);
     for (int i=2;i<=m;i++)
         if (p%i==0)
         {
            d=0;
            pd=1;
            while (p%i==0)
            {
                  d++;
                  p/=i;
                  pd*=i;
            }
            ans*=solve(i,d,pd);
         }
     if (p>1) ans*=solve(p,1,p);
     printf("%d\n",ans);
}
int main()
{
    int T;
    scanf("%d",&T);
    while (T--) solve();
}