[loj3330]猜数游戏

如果两个数$a_{x}$和$a_{y}$，$\exists 0<i,a_{x}^{i}\equiv a_{y}(mod\ p^{k})$，就建一条$x$到$y$的有向边，再对这张图强连通分量缩点，记$s_{i}$表示第$i$个点的大小，$f_{i}$表示能到达$i$的点（初始）数量，则答案为$\sum_{i=1}^{scc}\frac{2^{s_{i}}-1}{2^{s_{i}+f_{i}}}$（对每一个点分别统计贡献）

问题就变为如何判定$\exists i,x^{i}\equiv y(mod\ p^{k})$，分为$p\mid x$和$p\nmid x$两类

对于$x\mid p$和$y\nmid p$，$x^{i}\equiv y(mod\ p^{k})$的必要条件为$x^{i}\equiv y(mod\ p)$，而$x^{i}\equiv 0\ne y(mod\ p)$（交换$x$和$y$也同理可以证明），因此两类之间没有关系

对于$x\mid p$和$y\mid p$，必然有$\forall k\le i,x^{i}\equiv 0(mod\ p^{k})$，暴力枚举$i$即可，时间复杂度$o(n\log p)$

对于$x\nmid p$和$y\nmid p$，如果求出$且ord(x)=\min_{i>0且x^{i}\equiv 1(mod\ p^{k})}i$，那么$x^{i}\equiv y(mod\ p^{k})$当且仅当$ord(y)\mid ord(x)$

用BSGS求$ord(x)$复杂度及常数较大，考虑$ord(x)\mid \varphi(p)$，枚举$\varphi(p)$的质因子并判断能否消除，复杂度为$o(n\log^{2}p)$

#include<bits/stdc++.h>
using namespace std;
#define N 5005
#define mod 998244353
map<int,int>mat;
vector<int>v,v1[N],v2[N],v3[N];
int n,m,P,p,k,ans,a[N],ord[N],vis[N],dfn[N],sz[N],bl[N],dp[N];
int ksm(int n,int m,int p){
    if (!m)return 1;
    int s=ksm(n,m>>1,p);
    s=1LL*s*s%p;
    if (m&1)s=1LL*s*n%p;
    return s; 
}
int bsgs(int k){
    int ans=1;
    for(int i=0;i<v.size();i++)ans=ans*v[i];
    for(int i=0;i<v.size();i++)
        if (ksm(k,ans/v[i],P)==1)ans/=v[i];
    return ans;
}
void add(int x,int y){
    v1[x].push_back(y);
    v2[y].push_back(x);
}
void dfs1(int k){
    if (vis[k])return;
    vis[k]=1;
    for(int i=0;i<v1[k].size();i++)dfs1(v1[k][i]);
    dfn[++dfn[0]]=k;
}
void dfs2(int k){
    if (bl[k]){
        if (bl[k]!=m)v3[m].push_back(bl[k]);
        return;
    }
    sz[m]++;
    bl[k]=m;
    for(int i=0;i<v2[k].size();i++)dfs2(v2[k][i]);
}
void dfs3(int k,int x){
    if (vis[k])return;
    vis[k]=1;
    dp[x]+=sz[k];
    for(int i=0;i<v3[k].size();i++)dfs3(v3[k][i],x);
}
void init(){
    p=P;
    for(int i=3;i*i<=P;i++)
        if (P%i==0){
            p=i;
            break;
        }
    for(int i=P;i>1;i/=p)k++;
    int j=P/p*(p-1);
    for(int i=2;i*i<=j;i++)
        while (j%i==0){
            j/=i;
            v.push_back(i);
        }
    if (j>1)v.push_back(j);
}
void Kosaraju(){
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++)dfs1(i);
    for(int i=n;i;i--)
        if (!bl[dfn[i]]){
            m++;
            dfs2(dfn[i]);
        }
}
int main(){
    scanf("%d%d",&n,&P);
    init();
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        mat[a[i]]=i;
    }
    for(int i=1;i<=n;i++)
        if (a[i]%p)ord[i]=bsgs(a[i]);
        else{
            for(int j=1,t=a[i];t>0;j++,t=1LL*t*a[i]%P)
                if ((mat[t])&&(mat[t]!=i))add(i,mat[t]);
        }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if ((i!=j)&&(ord[i])&&(ord[j])&&(ord[i]%ord[j]==0))add(i,j);
    Kosaraju();
    for(int i=1;i<=m;i++){
        memset(vis,0,sizeof(vis));
        dfs3(i,i);
    }
    for(int i=1;i<=n;i++)ans=(ans+(ksm(2,sz[i],mod)-1LL)*ksm(2,n-dp[i],mod))%mod;
    printf("%d",ans);
    return 0;
}