HDU3836(tarjan+缩点)

/********************************************************************
* 题目分类：强连通+缩点
* 算法：tarjan
* 注意事项：数据范围大，要用邻接表，第一次没看范围果断爆栈了；
* problem:HDU3836-Equivalent Sets
* author：sgx
* date：2013/09/08
********************************************************************/
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <vector>
#include <algorithm>

#define A system("pause")

using namespace std;

const int maxn=20000+5;

vector <int>a[maxn];

int dfn[maxn],low[maxn],stack[maxn];
bool instack[maxn];

int indegree[maxn],outdegree[maxn];
int belong[maxn];

int top,Bcnt,index;

int m,n;

inline int min(int a,int b)
{
    return a<b?a:b;
}

inline int max(int a,int b)
{
    return a>b?a:b;
}

void tarjan(int u)
{
     int v;
     dfn[u]=low[u]=++index;
     stack[++top]=u;
     instack[u]=true;

     for(int i=0;i<a[u].size();i++)//枚举每条边；
     {
        int v=a[u][i];//找到邻接点；
        if(!dfn[v])//若未访问过则访问该点；
        {
             tarjan(v);
             low[u]=min(low[u],low[v]);//树边；
        }
        else if(instack[v])
        {
             low[u]=min(low[u],dfn[v]);//后向边；
        }
     }

     if(dfn[u]==low[u])
     {
         Bcnt++;
         do
         {
             v=stack[top--];//弹出栈顶元素；
            belong[v]=Bcnt;//标记强连通分量；
            instack[v]=false;
         }while(v!=u);
     }
}

void solve()
{
     int i;

     top=index=Bcnt=0;
     memset(instack,0,sizeof(instack));
     memset(dfn,0,sizeof(dfn));
     memset(low,0,sizeof(low));
     memset(indegree,0,sizeof(indegree));
     memset(outdegree,0,sizeof(outdegree));

     for(i=1;i<=n;i++)
     {
        if(!dfn[i])//如果某点没被访问过，则对其进行tarjan;
         tarjan(i);//tarjan结果是得到了belong数组,记录每个节点分别属于哪个强联通分量
     }

     for(int i=1;i<=n;i++)
     {
         for(int j=0;j<a[i].size();j++)
         {
            int k=a[i][j];
            if(belong[i]!=belong[k])//两点之间有边，但不是属于一个强联通分量的边
            {
                 outdegree[belong[i]]++; // 缩点后的点入度+1
                 indegree[belong[k]]++;// 缩点后的点出度+1

            }
         }
    }
    int cnt1=0,cnt2=0;

     for(int i=1;i<=Bcnt;i++)
     {
        if(!indegree[i])
             cnt1++;
        if(!outdegree[i])
             cnt2++;
     }
     if(Bcnt==1)
         printf("0\n");
     else
         printf("%d\n",max(cnt1,cnt2));
}

int main()
{
     while(scanf("%d%d",&n,&m)!=EOF)
     {
         for(int i=0;i<=maxn;i++)
             a[i].clear();
         for(int i=1;i<=m;i++)
         {
            int u,v;
             scanf("%d%d",&u,&v);
             a[u].push_back(v);
         }

         solve();
     }
     return 0;
}