tarjan强连通缩点——cf711D

模板题

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define N 500005
#define mod 1000000007
int n,a[N];
vector<int>G[N];
int cnt,dfn[N],low[N],ind,stk[N],ins[N],top;
vector<int>scc[N];
void tarjan(int x){
    dfn[x]=low[x]=++ind;
    stk[++top]=x,ins[x]=;
    for(int i=;i<G[x].size();i++){
        int y=G[x][i];
        if(!dfn[y]){
            tarjan(y);
            low[x]=min(low[x],low[y]);
        }
        else if(ins[y])
            low[x]=min(low[x],low[y]);
    }
    if(dfn[x]==low[x]){
        cnt++;int y;
        do{
            y=stk[top--];
            ins[y]=;
            scc[cnt].push_back(y);
        }while(y!=x);
    }
}
ll Pow(ll a,ll b){
    ll res=;
    while(b){
        if(b%)
            res=res*a%mod;
        b>>=;a=a*a%mod;
    }
    return res;
}
int main(){
    cin>>n;
    for(int i=;i<=n;i++){
        cin>>a[i];
        G[i].push_back(a[i]);
    }
    for(int i=;i<=n;i++)
        if(!dfn[i])
            tarjan(i);
 
    ll ans=;
    for(int i=;i<=cnt;i++)
        if(scc[i].size()!=)
            ans=ans*(Pow(,scc[i].size())-+mod)%mod;
        else ans=ans*%mod;
    cout<<ans<<'\n';
}