洛谷P2002 消息扩散（tarjan）

消息扩散

题目描述

有n个城市，中间有单向道路连接，消息会沿着道路扩散，现在给出n个城市及其之间的道路，问至少需要在几个城市发布消息才能让这所有n个城市都得到消息。

输入输出格式

输入格式：
第一行两个整数n,m表示n个城市，m条单向道路。

以下m行，每行两个整数b,e表示有一条从b到e的道路，道路可以重复或存在自环。

输出格式：
一行一个整数，表示至少要在几个城市中发布消息。

分析：用tarjan缩点，入度为0的强连通分量的个数就是ans。

代码

#include <cstdio>
#include <cstring>
#include <stack>
#define maxn 600000
using namespace std;
stack<int> s;
struct edge
{
    int x,y,nxt;
}g[maxn];

int p=0,n,m,ans=0;
bool instack[maxn];
int dfn[maxn],low[maxn],ls[maxn],belong[maxn],d[maxn];

void tarjan(int i)
{
    p++;
    s.push(i);
    instack[i]=true;
    dfn[i]=low[i]=p;
    for (int k=ls[i];k;k=g[k].nxt)
    {
        int j=g[k].y;
        if (dfn[j]==0)
        {
            tarjan(j);
            if (low[j]<low[i])
                low[i]=low[j];
        }
        else 
        {
            if (instack[j]&&dfn[j]<low[i])
                low[i]=dfn[j];
        }
    }
    if (dfn[i]==low[i])
    {
        ans++;
        int j;
        do
        {
            j=s.top();
            s.pop();
            instack[j]=false;
            belong[j]=ans;
        }
        while (i!=j);
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d",&g[i].x,&g[i].y);
        g[i].nxt=ls[g[i].x];
        ls[g[i].x]=i;
    }
    for (int i=1;i<=n;i++)
        if (dfn[i]==0) tarjan(i);
    for (int i=1;i<=m;i++)
        if (belong[g[i].x]!=belong[g[i].y])
            d[belong[g[i].y]]++;
    int o=0;
    for (int i=1;i<=ans;i++)
        if (d[i]==0) o++;
    printf("%d",o);
}