tarjan缩点以及链式前向星的基本+应用（洛谷1262 间谍网络）

  1 #include<bits/stdc++.h>
  2 #define maxn 22000
  3 #define inf 29999999
  4 using namespace std;
  5 
  6 int n,p,r;
  7 int val[maxn];
  8 int cnt=0;
  9 int peo[maxn][2];
 10 int book[maxn];
 11 int head [maxn]; 
 12 
 13 int dfn[maxn],low[maxn];
 14 int cn=0;
 15 stack<int> s;
 16 int color=0;
 17 int colored[maxn];
 18 
 19 int input[maxn];//入度 
 20 int cost[maxn];
 21 
 22 struct Edge
 23 {
 24    int to,next,w;
 25 };
 26 struct Edge edge[maxn];
 27 
 28 void add(int x,int y)
 29 {
 30    edge[++cnt].to=y;
 31    edge[cnt].next=head[x];
 32    head[x]=cnt;
 33 }
 34 
 35 int BFS()
 36 {
 37     queue<int >  q;
 38     for(int i=1;i<=p;i++)
 39       {
 40          q.push(peo[i][0]);
 41          book[peo[i][0]]=1;
 42       }
 43       while(!q.empty())
 44       {
 45            int x=q.front();
 46            q.pop();
 47            for(int i=head[x];i;i=edge[i].next)
 48            {
 49               int j=edge[i].to;
 50               if(book[j]==0)
 51               {
 52                   book[j]=1;
 53                   q.push(j);
 54               }
 55            }
 56       }
 57       for(int i=1;i<=n;i++)
 58         if(!book[i])
 59            return i;
 60     return 0;
 61     
 62 }
 63 
 64 
 65 void tarjan(int x)
 66 {
 67     dfn[x]=low[x]=++cn;
 68     book[x]=1;
 69     s.push(x);
 70     
 71     int j;
 72     for(int i=head[x];i;i=edge[i].next)
 73     {
 74           j=edge[i].to;
 75           if(!dfn[j])
 76           {
 77                tarjan(j);
 78                low[x]=min(low[x],low[j]);
 79           }
 80           else if(book[j])
 81              low[x]=min(low[x],dfn[j]);
 82     }
 83     
 84     if(dfn[x]==low[x])
 85     {
 86          color++;
 87          do
 88          {
 89               j=s.top();
 90               s.pop(); 
 91               colored[j]=color;
 92               book[j]=0;
 93              
 94          }while(x!=j);
 95     }
 96     return ;
 97 }
 98 
 99 void solve()
100 {
101     memset(book,0,sizeof(book));
102     for(int i=1;i<=n;i++)
103       if(!dfn[i])
104          tarjan(i);
105     for(int i=1;i<=n;i++)
106     {
107         for(int j=head[i];j;j=edge[j].next)
108         {
109            int to=edge[j].to;
110            if(colored[to]!=colored[i])
111            {
112                 input[colored[to]]++;
113             }
114         }
115     }
116     for(int i=1;i<=n;i++)
117        cost[i]=inf;
118     for(int i=1;i<=p;i++)
119        cost[colored[peo[i][0]]]=min(cost[colored[peo[i][0]]],peo[i][1]);
120      int ans_last=0;
121     for(int i=1;i<=color;i++)
122        if(input[i]==0)
123            ans_last+=cost[i];
124     cout<<"YES\n"<<ans_last<<endl;
125     return ;
126 }
127 
128 int main()
129 {
130     ios::sync_with_stdio(false);
131     cin>>n;
132     cin>>p;
133     int t1,t2;
134     for(int i=1;i<=p;i++)
135          cin>>peo[i][0]>>peo[i][1];
136     cin>>r;
137     for(int i=1;i<=r;i++)
138     {
139         cin>>t1>>t2;
140         add(t1,t2);        
141     }
142     int ans=BFS();
143     if(ans)
144     {
145         cout<<"NO\n"<<ans<<endl;
146         return 0;
147     }
148     solve();
149     return 0;
150 }