Codeforces Round #369 (Div. 2) D. Directed Roads （图论+组合数学+dfs）

题目链接


思路：一开始想的太复杂，其实它只有n条边的话，它的图其实不会很复杂，最多是多个环加一条链，不会有环套环地情况，对于一个环地所有边来说除了不变和全部一起变两种情况外，其他地变化都可以，所有环地贡献为pow（2，n）-2，n为环的边数，对于非环边，它要么变要么不变，就有pow(2,n)的贡献。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const int maxn=2e5+1; 
int n,a[maxn];
ll sum,ans=1,vis[maxn],dep[maxn];
ll quick(ll a,ll b){ 
    ll ans=1;  
    a=a%mod;
    while(b!=0){
        if(b&1) ans=(ans*a)%mod;
        b>>=1;
        a=(a*a)%mod;
    }
    return ans;
}
void dfs(int u,int deep,int fa)
{
	 vis[u]=fa;
	 dep[u]=deep;
	 if(!vis[a[u]]) dfs(a[u],deep+1,fa);
	 else if(vis[a[u]]==fa)
	 {
	 	ans=ans*(quick(2,dep[u]-dep[a[u]]+1)-2+mod)%mod;
	 	sum+=dep[u]-dep[a[u]]+1;
	 }
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;++i) scanf("%d",&a[i]);
	for(int i=1;i<=n;++i) if(!vis[i]) dfs(i,0,i);
	printf("%lld\n",(ans*(quick(2,n-sum)%mod)%mod));
}