Endless Walk(ABC 245 F)

AtCoder - abc245_f

题目大意

你有一张\(N\)个点，\(M\)条边的简单有向连通图\(G\)，第\(i\)条边从\(U_i\)指向\(V_i\)，问有多少个这样的点使得你从这个点出发可以无穷无尽的走下去。\((1\le N\le2\times10^5,0\le M\le\min(N(N-1),2\times10^5))\)

思路

有一个类似拓扑的思想，就是说对于每一个初度为\(0\)的点，它都不可能无穷无尽地走下去，那么我们减去这样的点，就是我们要的答案了，这题就做完了。

代码

#include<bits/stdc++.h>
using namespace std;
int out[200005];
int vis[200005];
vector<int>edge[200005];
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        out[u]++;
        edge[v].push_back(u);
    }
    queue<int>q;
    int ans=n;
    for(int i=1;i<=n;i++)if(!out[i])
    {
        q.push(i);
        vis[i]=1;
        ans--;
    }
    while(!q.empty())
    {
        int p=q.front();
        q.pop();
        for(auto it:edge[p])if(!vis[it])
        {
            out[it]--;
            if(!out[it])
            {
                q.push(it);
                vis[it]=1;
                ans--;
            }
        }
    }
    printf("%d\n",ans);
    return 0;
}