P2341 [HAOI2006]受欢迎的牛

· · 个人记录

题目描述

每头奶牛都梦想成为牛棚里的明星。被所有奶牛喜欢的奶牛就是一头明星奶牛。所有奶

牛都是自恋狂，每头奶牛总是喜欢自己的。奶牛之间的“喜欢”是可以传递的——如果A喜

欢B，B喜欢C，那么A也喜欢C。牛栏里共有N 头奶牛，给定一些奶牛之间的爱慕关系，请你

算出有多少头奶牛可以当明星。

输入输出格式

输入格式：

 第一行：两个用空格分开的整数：N和M

 第二行到第M + 1行：每行两个用空格分开的整数：A和B，表示A喜欢B

输出格式：

 第一行：单独一个整数，表示明星奶牛的数量

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 1e4 + 4;
const int maxm = 5e4 + 4;

int n,m,u,v;
int head[maxn],tot;
int dfn[maxn],low[maxn],ind;
int stack[maxn],top,num[maxn],belong[maxn],cnt;
bool ins[maxn];
int chu[maxn],ans;
struct Edge{
    int from,to,next;
}edge[maxm];

void add(int u,int v){
    edge[++tot].from = u;
    edge[tot].to = v;
    edge[tot].next = head[u];
    head[u] = tot;
}

int read(){
    char ch = getchar();
    int f = 1 , x = 0;
    while(ch > '9' || ch < '0'){
        if(ch == '-')  f = -1;
        ch = getchar();
    }
    while(ch >= '0' && ch <= '9'){
        x = x * 10 + ch - '0';
        ch = getchar(); 
    }
    return x * f;
}

void tarjan(int x){
    dfn[x] = low[x] = ++ind;
    stack[++top] = x;
    ins[x] = true;
    for(int i=head[x];i;i=edge[i].next){
        int v = edge[i].to;
        if(ins[v]) low[x] = min(low[x] , dfn[v]);
        if(!dfn[v]){
            tarjan(v);
            low[x] = min(low[x] , low[v]);
        }
    }
    int k = 0;
    if(dfn[x] == low[x]){
        cnt++;
        do{
            k = stack[top];
            num[cnt]++;
            top--;
            ins[k] = false;
            belong[k] = cnt;
        } while(k != x);
    }
} 

int main(){
    n = read();  m = read();
    for(int i=1;i<=m;i++){
        u = read();  v = read();
        add(u , v);
    }
    for(int i=1;i<=n;i++)
      if(!dfn[i])  tarjan(i);
    for(int i=1;i<=m;i++){
        if(belong[edge[i].from] != belong[edge[i].to]){
            chu[belong[edge[i].from]]++;
        }
    }
    int sum = 0;
    for(int i=1;i<=cnt;i++)
      if(!chu[i])  ans = num[i] , sum++;
    if(sum >= 2)  printf("0");
    else printf("%d",ans);
    return 0;
}