Directed Roads （dfs 组合数学 图论）

题目

题意：给一张N个点N条有向边的图，边可以逆向。问任意逆向若干条边使得这张图无环的方案数(mod 1e9+7)。

思路： 只有两种情况反边不能消除环，处理全部不反向和全部反向两种情况..
对于每一个环，我们都有2^n-2种方案使得这个环不存在，就是2^n减去全部翻转，和全部不翻转的方案。
最后再处理不在环上的边的数量就可以了。

input
3
2 3 1
output
6

#include<cstdio>
#include<cstring>
#include<stack>
#define ll long long
#define mod 1000000007
using namespace std;
int a[220000];
int vis[220000];
int dep[220000];
ll mi[220000];
int dis;
stack<int> st;
struct node
{
    int to,nnext;
}edge[220000];
int pre[220000],cnt;
int time;
void init()
{
    cnt=0;
    time=0;
    memset(pre,-1,sizeof(pre));
}
void add_edge(int u,int v)
{
    edge[cnt].to=v;
    edge[cnt].nnext=pre[u];
    pre[u]=cnt++;
}
void dfs(int u)
{
   vis[u]=time;
   st.push(u);
   for(int i=pre[u];i!=-1;i=edge[i].nnext)
   {
       int v=edge[i].to;
       if(!vis[v]) dfs(v);
       else if(vis[v]==time)
       {
           int t=st.top();
           dis++;

           st.pop();
           while(t!=v)
           {
               t=st.top();
               st.pop();
               dis++;

           }

       }

   }
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        init();
        mi[0]=1;
        for(int i=1;i<220000;i++)
            mi[i]=(mi[i-1]*2)%mod;
        while(!st.empty()) st.pop();
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            add_edge(i,a[i]);
        }
        memset(vis,0,sizeof(vis));
        memset(dep,0,sizeof(dep));
        ll ans=1;
        int ydis=n;
        for(int i=1;i<=n;i++)
        {
            if(!vis[i])
            {
            time++;
            dis=0;
             dfs(i);

             if(dis>=2)
             ans=ans*(mi[dis]-2+mod)%mod;

             ydis-=dis;
            }
        }

        ans=ans*mi[ydis]%mod;
        printf("%I64d\n",ans);
    }
    return 0;
}