Mouse Hunt

Mouse Hunt

给定一个n个点的图，每个点有权值\(c_i\)，并且只有一条出边。现在你要在一些点上打标记，使得从任何一个点出发最终都会经过有标记的点。求标记点的权值和最小值。

就是找环啊！拓扑排序啊！

#include <cstdio> 
#include <algorithm>
using namespace std;

const int maxn=2e5+5;
int n, c[maxn], a[maxn], in[maxn], vis[maxn];
int q[maxn], h, t, ans;

int dfs(int now){
    vis[now]=1;
    if (vis[a[now]]) return c[now];
    return min(c[now], dfs(a[now]));
}

int main(){
    scanf("%d", &n); int now;
    for (int i=1; i<=n; ++i) scanf("%d", &c[i]);
    for (int i=1; i<=n; ++i){ scanf("%d", &a[i]); ++in[a[i]]; }
    for (int i=1; i<=n; ++i) if (!in[i]) q[t++]=i;
    while (h<t){
        now=q[h++]; --in[a[now]];
        if (!in[a[now]]) q[t++]=a[now];
    }
    for (int i=1; i<=n; ++i) if (in[i]&&!vis[i]) ans+=dfs(i);
    printf("%d\n", ans);
    return 0;
}