luogu1262 间谍网络（tarjan缩点）

tarjan缩点，就是用来把一个有向带环图变成DAG即 Directed Acyclic Graph有向无环图的算法，复杂度为O(E+V)。dfn[i]是dfs序，low[i]是i及i的子树中最小的dfn。belong[x]是点x在新图中所属的点。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 3010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
struct edge{
    int to,next;
}data[8010];
int n,m,w[N],h[N],num=0,ans=0,tot=0,sw[N];
int dfn[N],low[N],num1=0,belong[N],rd[N];
stack<int>q;
bool vis[N],inq[N];
void dfs(int x){
    if(vis[x]) return;
    vis[x]=1;tot++;
    for(int i=h[x];i;i=data[i].next) dfs(data[i].to);
}
void tarjan(int x){
    dfn[x]=low[x]=++num1;
    q.push(x);inq[x]=1;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;
        if(!dfn[y]){
            tarjan(y);
            low[x]=min(low[x],low[y]);
        }
        else if(inq[y]) low[x]=min(low[x],dfn[y]);
    }
    if(dfn[x]==low[x]){
        sw[x]=inf;
        while(1){
            int y=q.top();q.pop();inq[y]=0;
            belong[y]=x;sw[x]=min(sw[x],w[y]);
            if(y==x) break;
        }
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();memset(w,0x3f,sizeof(w));
    while(m--){
        int x=read(),y=read();w[x]=y;
    }m=read();
    while(m--){
        int x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
    }
    for(int i=1;i<=n;++i) if(w[i]!=inf) dfs(i);//是否能遍历所有点 
    if(tot<n) 
        for(int i=1;i<=n;++i) 
            if(!vis[i]){printf("NO\n%d",i);return 0;}
    for(int i=1;i<=n;++i) if(w[i]!=inf&&!dfn[i]) tarjan(i);
    for(int x=1;x<=n;++x){
        for(int i=h[x];i;i=data[i].next){
            int y=data[i].to;
            if(belong[x]!=belong[y]) rd[belong[y]]++;
        }
    }
    for(int i=1;i<=n;++i)//没有入度，也就是说没有别的点可以到这里，只能收买。 
        if(belong[i]==i&&!rd[i]) ans+=sw[i];
    printf("YES\n%d",ans);
    return 0;
}