Codeforces 711D dfs找环

题目：http://codeforces.com/contest/711/problem/D

题意：

有一个n个点的有向图，有n条边，分别从每个点 i 出发指向某个点ai，现在可以把某些边翻转，问总共可以得到多少种无换图？
例如输入：
3
2 3 1
一共有6种方案

分析：

要想得到无换图，那么就需要把环找出来，然后至少翻转一条边就可以了，总的方法数是2^x-2,2是一条边也不翻转和翻转所有边。
其他不在环中的边可以翻转也可以不翻转，所以方案数是2^x

#include<bits/stdc++.h>
using namespace std;
#define rep(i,s,t) for(int i=(s);i<(t);i++)
#define per(i,t,s) for(int i=(t);i>=(s);i--)
#define fi first
#define se second
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int>pii;
const int mod = 1e9 + 7;
const int N = 2e5 + 9;
vector<pii>g[N];
int r;
int d[N], vis[N];
void dfs (int u, int fa, int deep) {
    if (vis[u]) {
        if (r == 0) r = deep - d[u];
        return;
    }
    vis[u] = 1;
    d[u] = deep;
    rep (i, 0, g[u].size() ) {
        if (g[u][i].se == fa) continue;
        dfs (g[u][i].fi, g[u][i].se, deep + 1);
    }
}
ll f[N];
int main() {
    //freopen("f.txt","r",stdin);
    int n, x;
    scanf ("%d", &n);
    f[0] = 1;
    rep (i, 1, n + 1) f[i] = f[i - 1] * 2 % mod;
    rep (i, 1, n + 1) {
        scanf ("%d", &x);
        g[i].push_back (pii (x, i) );
        g[x].push_back (pii (i, i) );
    }
    ll ans = 1;
    int sum = 0;
    rep (i, 1, n + 1) if (!vis[i]) {
        r = 0;
        dfs (i, 0, 0);
        sum += r;
        //cout<<r<<endl;
        ans = ans * (f[r] - 2) % mod;
    }
    cout << ans* (f[n - sum]) % mod << endl;
    return 0;
}
/*
Input

3
2 3 1

Output

6
*/