tarjan+缩点（poj2186）

题目链接：http://poj.org/problem?id=2186

思路：tarjan缩点后找入出度为0的点。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <bitset>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <sstream>
#include <iomanip>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%I64d",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod int(1e9+7)
#define pb push_back
#define lc (d<<1)
#define Pll pair<ll,ll>
#define P pair<int,int>
#define pi acos(-1)
int deep,dfn[10008],low[10008],vis[10008],z[10008],d[10008],e[10008],sum,tmp,n,m;
vector<int>g[10008];
stack<int> st;
void tarjan(int u)
{
    dfn[u]=low[u]=++deep;
    vis[u]=1,st.push(u);
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(!dfn[v])
        {
            tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        else if(vis[v])  low[u]=min(low[u],dfn[v]);
    }
    if(dfn[u]==low[u])
    {
        sum++,tmp=0;
        while(tmp!=u)
	{
            tmp=st.top();st.pop();
            z[tmp]=sum;
            e[sum]++;
            vis[tmp]=0;
	}
    }
}
int main()
{
    cin.tie(0);
    cout.tie(0);
    while(cin>>n>>m)
    {
        int x,y;
        FOR(i,1,n)  g[i].clear();
        while(m--)
        {
            si(x),si(y);
            g[x].pb(y);
        }
        sum=deep=0;
        if(st.size())  st.pop();
        REW(vis,0),REW(dfn,0),REW(low,0);
        FOR(i,1,n)  if(!dfn[i])  tarjan(i);
        int s=0,qw=0;
        FOR(i,1,n)
        {
            for(int j=0;j<g[i].size();j++)
            {
                if(z[i]!=z[g[i][j]])  d[z[i]]++;
            }
        }
        FOR(i,1,sum)
        {
            if(!d[i])  qw++,s=e[i];
        }
        if(qw!=1)  puts("0");
        else  cout<<s<<endl;
    }
    return 0;
}