poj 2186 Popular Cows 有向图强连通分量 tarjan

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
//#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100000+5;
struct Edge
{
int s,t;
};
vector<Edge>edge;
stack<int>st;
vector<int>g[100000+5];
int n,m,in
,id
,dfn
,low
,def,cnt;

void Addedge(int u,int v)
{
Edge tp;
tp.s=u,tp.t=v;
edge.push_back(tp);
g[u].push_back(edge.size()-1);
}

void dfs(int u)
{
int k,mx=dfn[u]=low[u]=++def;
st.push(u);
for(int i=0; i<g[u].size(); i++)
{
int v=edge[g[u][i]].t;
if(dfn[v]==-1) dfs(v);
if(low[v]<mx) mx=low[v];
}
if(mx<low[u])
{
low[u]=mx;
return ;
}
do
{
k=st.top();
id[k]=cnt;
low[k]=n;
st.pop();
}
while(k!=u);
cnt++;
}
int main()
{
int i,j,u,v,ans,pos;
//freopen("in.txt","r",stdin);
while(~scanf("%d%d",&n,&m))
{
for(i=0; i<n; i++)
{
g[i].clear();
id[i]=i;
in[i]=0;
}
edge.clear();
for(i=0; i<m; i++)
{
scanf("%d%d",&u,&v);
Addedge(u-1,v-1);
}
while(!st.empty()) st.pop();
def=0;
cnt=0;
memset(dfn,0xff,sizeof(dfn));
for(i=0; i<n; i++)
{
if(dfn[i]==-1)
dfs(i);
}
for(i=0; i<m; i++)
{
u=id[edge[i].s];
v=id[edge[i].t];
if(u!=v) in[u]++;
//printf("%d %d\n",u,v);
}
ans=0;
pos=-1;
for(i=0; i<cnt; i++)
{
if(in[i]==0)
{
//printf("%d\n",id[i]);
if(pos==-1)
{
ans++;
pos=i;
}
else if(pos!=i) ans++;
}
}
// printf("%d\n",pos);
if(ans!=1) printf("0\n");
else
{
ans=0;
for(i=0; i<n; i++)
{
if(id[i]==pos)
ans++;
}
printf("%d\n",ans);
}
}
return 0;
}