HDU 5326 多维数组接法 【并查集】

A - Work

Time Limit: 1000 MS      Memory Limit: 32768 KB      64bit IO Format: %I64d & %I64u

Description

It’s an interesting experience to move from ICPC to work, end my college life and start a brand new journey in company. 

As is known to all, every stuff in a company has a title, everyone except the boss has a direct leader, and all the relationship forms a tree. If A’s title is higher than B(A is the direct or indirect leader of B), we call it A manages B. 

Now, give you the relation of a company, can you calculate how many people manage k people. 

 

Input

There are multiple test cases. 

Each test case begins with two integers n and k, n indicates the number of stuff of the company. 

Each of the following n-1 lines has two integers A and B, means A is the direct leader of B. 

1 <= n <= 100 , 0 <= k < n 

1 <= A, B <= n 

 

Output

For each test case, output the answer as described above.

 

Sample Input

       
       

        
        

         
         7 2
        
        
1 2

        
        

         
         2 4
        
        
1 3
2 5

        
        

         
         3 7 
        
        
        
        

         
         3 6
        
        
       
       

 

Sample Output

       
       

        
        2 
       
       

 

题目就是给了一堆数，分别对应一个人，然后吧就是A领导B，最后让你求出有K个手下的人的个数。（开始翻译成了求K的手下有几个。。英语不好真苦逼啊）。。。

本题其实是一道并查集的题目，但是并查集并没有写明白，所以用了苦逼的多维数组。。。

详情请看代码。。。

然后开始贴代码

#include <iostream>//本人AC代码  
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
int has[2][105][105];  //表示领导关系的成立与否 成立为1 不成立默认是0
int son[105];               //表示第几个人的手下
int n,m,sum;
int  fin(int x)              //核心代码
{
    int num=0;
        for(int j=1; j<=n; j++)
        {
            if(has[1][x][j]==1&&has[0][x][j]==0)
            {
                num++;
                has[0][x][j]=1;
                son[x]+=fin(j);
            }
        }
    return son[x];
}
int main ()
{
    while(~scanf("%d%d",&n,&m))
    {
        memset(has,0,sizeof(has));//数组清0
        memset(son,0,sizeof(son));
        int a,aa;
        sum=0;
        for(int i=1; i<n; i++)
        {
            scanf("%d%d",&a,&aa);
            has[1][a][aa]=1;      //表示领导关系
            son[a]++;                 //这里把a的直接手下记录
        }
        for(int i=1; i<=n; i++)   // 这里来计算每个人的间接手下
        {
            for(int j=1; j<=n; j++)
            {
                if(has[1][i][j]==1&&has[0][i][j]==0)   //后面的那个用来标记遍没遍历，不知道到底有没有用 但是这样比较保险
                {
                    has[0][i][j]=1;
                    son[i]+=fin(j);
                }
            }
        }
        //测试的代码   
        /* 
        for(int i=1; i<=n; i++)
        {
            printf("**%d\n",son[i]);
        }
        */
        for(int i=1; i<=n; i++)
        {
            if(m==son[i])
                sum++;
        }
        printf("%d\n",sum);
    }
}

另附几组测试数据

7 6  //1
1 2
1 3
2 4
2 5
3 6
3 7


7 2  //2
1 2
1 3
2 4
2 5
3 6
3 7


7 2 // 2
7 6
7 5
6 4
6 3
5 2
5 1


7 1  //1
1 2
2 3
3 4
4 5
5 6
6 7