【杭电】[5326]Work - BoilTask's Blog

问题描述

问题分析

寻找父节点时则把父节点的层数加1
这样ran数组中便可保存其下面有多少人
最后查找ran[i]==k的个数即可

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

#include<stdio.h>
int par[120],ran[120];
void find(int m) {
    if(m!=par[m]) {
        ran[par[m]]++;
        find(par[m]);
    }
}
void unite(int x,int y) {
    if(x!=y)
        par[y]=x;
}
int main() {
    int n,k;
    while(scanf("%d %d",&n,&k)!=EOF) {
        for(int i=1; i<=n; i++) {
            ran[i]=0;
            par[i]=i;
        }
        for(int i=1; i<n; i++) {
            int a,b;
            scanf("%d %d",&a,&b);
            unite(a,b);
        }
        for(int i=1; i<=n; i++)
            find(i);
        int res=0;
        for(int i=1; i<=n; i++) {
            if(ran[i]==k)
                res++;
        }
        printf("%d\n",res);
    }
    return 0;
}