Targin缩点

#include<bits/stdc++.h>
using namespace std;

const int N = 10010, M = 50010;

int n, m;
int h[N], w[M], e[M], ne[M], idx;
int dfn[N], low[N], timestamp;
int stk[N], top;
bool in_stk[N];
int id[N], scc_cnt, size_scc[N];
int dout[N];// 记录每个连通分量的出度

void add(int a, int b){
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}

void tarjan(int u){
    //当前点的时间戳
    dfn[u] = low[u] = ++ timestamp;
    // 加入栈中
    stk[++ top] = u, in_stk[u] = true;

    //遍历u点的所有邻点
    for(int i = h[u]; ~i; i = ne[i]){
        int j = e[i];
        if(!dfn[j]){//如果没有遍历过
            tarjan(j); // 遍历它
            low[u] = min(low[u], low[j]);
        }
        // 当前点在栈当中
        else if(in_stk[j]) low[u] = min(low[u], dfn[j]);
    }

    if(dfn[u] == low[u]){
        ++ scc_cnt; // 更新强连通分量的编号
        int y;
        do{
            y = stk[ top--]; //不断取出栈内元素
            in_stk[y] = false;
            id[y] = scc_cnt; //y元素所属的连通块编号
            size_scc[scc_cnt] ++; //该连通块内包含的点数
        }while(y != u); // 直到y不等于u
    }

}

int main(){
    cin >> n >> m;
    memset(h, -1, sizeof h);
    while(m --){
        int a, b;
        cin >> a >> b;
        add(a, b);
    }

    for(int i = 1; i <= n; i ++){
        if(!dfn[i])
            tarjan(i);
    }

    // 建有向无环图
    // 统计在新图中所有点的出度
    for(int i = 1; i <= n; i ++){
        for(int j = h[i]; ~j; j = ne[j]){
            int k = e[j];
            int a = id[i]; //a表示i所在连通分量的编号
            int b = id[k]; // b表示k所在连通分量的编号
            //如果点i和点k不在同一个连通块
            // dout存的是出度，因为本题只需要出度
            //在其他题目中，可能是要建边，因为这里是构造有向无环图
            if(a != b) dout[a] ++; // 从a走到b，a的出度++
        }
    }

    // 和本题有关的部分：
    // zeros是统计在新图中，出度为0的点的个数
    // sum表示满足条件的点（最受欢迎的奶牛）的个数
    int zeros = 0, sum = 0;
    for(int i = 1; i <= scc_cnt; i ++){
        if(!dout[i]){
            zeros ++;
            sum += size_scc[i];
            if(zeros > 1){
                sum = 0;
                break;
            }
        }
    }
    cout << sum << endl;
}