P2002 消息扩散

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<stack>
using namespace std;
#define maxn 10101110
stack<int>q;
struct Node
{
    int next,to;
}e[maxn],ee[maxn];
int dfn[maxn],low[maxn],be[maxn];
int vis[maxn],in[maxn],headd[maxn];
int n,m,num,num1,ans,head[maxn];
//num1表示大点的个数，num表示tarjan中的编号
//head原图，e->原图；ee->新图 headd->新图
void add(int x,int y)
{
    e[++head[0]].next=head[x];
    e[head[0]].to=y;
    head[x]=head[0];
}
void ad(int x,int y)
{
    ee[++headd[0]].next=headd[x];
    ee[headd[0]].to=y;
    headd[x]=headd[0];
}
//这里我用的head[0]，headd[0]表示是为了节省一个小小的空间
void tarjan(int x)
{
    vis[x]=1;q.push(x);
    dfn[x]=low[x]=++num;
    for(int i=head[x];i;i=e[i].next)
    {
        int t=e[i].to;
        if(!dfn[t])
        {
            tarjan(t);
            low[x]=min(low[x],low[t]);
        }
        else if(vis[t])
            low[x]=min(low[t],low[x]);
    }
    if(dfn[x]==low[x])
    {
        vis[x]=0;
        be[x]=++num1;
        while(q.top()!=x)
        {
            be[q.top()]=num1;
            vis[q.top()]=0;
            q.pop();
        }
        be[q.top()]=num1;
        vis[q.top()]=0;
        q.pop();
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int y,u;
        scanf("%d%d",&y,&u);
        add(y,u);
    }//加边
    for(int i=1;i<=n;i++)
        if(!dfn[i])
            tarjan(i);
    for(int i=1;i<=n;i++)
        for(int j=head[i];j;j=e[j].next)
            if(be[i]!=be[e[j].to])
            {
                ad(be[i],be[e[j].to]);
                in[be[e[j].to]]++;
                //加新边同时记in
            }
    for(int i=1;i<=num1;i++)
        if(!in[i])
            ans++;
    printf("%d",ans);
    return 0;
}