AcWing 1977. 信息中继

题意

对于奶牛 i，它会将自己收到的任何信息转发给奶牛 Fi（Fi 与 i 一定不同），如果 Fi 为 0，则奶牛 i 不转发消息；判断有多少头奶牛发送的信息不会陷入死循环。

分析

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 1010;

int n, a[N], p[N], s[N];

int find(int x){
    if(p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main()
{
    scanf("%d", &n);
    for(int i = 1; i <= n; i ++) p[i] = i, s[i] = 1;
    for(int i = 1; i <= n; i ++){
        scanf("%d", &a[i]);
        if(a[i]){
            if(find(i) != find(a[i])){
                s[find(a[i])] += s[find(i)];
                p[find(i)] = p[find(a[i])];
            }
        }
    }
    int ans = 0;
    for(int i = 1; i <= n; i ++){
        if(find(i) == i && a[i] == 0){
            ans += s[i];
        }
    }
    printf("%d\n", ans);
    
    return 0;
}