luogu1262 间谍网络(tarjan缩点)

来源:互联网 发布:粉红女郎 知乎 编辑:程序博客网 时间:2024/05/26 19:14

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 3010inline 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;}