hdu(5326)——Work

题意：

现在有一个有n个人的公司，然后每个人除了boss外都有一个直接领导者，然后如果是直接或间接领导的话，那么都可以说是A管理了B。然后问你有多少个人管理着k个人。

思路：

1）第一种方法：

我一开始就想到的是并查集，但是这里并不仅仅是简单的并查集。我们需要保存下来每个人与其他人的关系，要不然的话，我们无法直接合并（因为这里是有方向的）。

所以我们记vis[x][y]为x是y的领导。然后三层for更新每个人之间的关系。

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 111
int par[maxn],num[maxn],a[maxn],b[maxn];
int vis[maxn][maxn];
int n,k;
void init(){
    for(int i=1;i<=n;i++){
        par[i]=i;
        num[i]=0;
    }
}
int find(int x){
    if(par[x]==x) return par[x];
    else par[x]=find(par[x]);
}
void merge(int x,int y){
    int tx=find(x);
    int ty=find(y);
    if(tx!=ty){
        par[ty]=tx;
    }
}
int main(){
    while(~scanf("%d%d",&n,&k)){
        init();
        memset(vis,0,sizeof(vis));
        for(int i=1;i<n;i++){
            scanf("%d%d",&a[i],&b[i]);
            merge(a[i],b[i]);
            vis[a[i]][b[i]]=1;
        }
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                for(int l=1;l<=n;l++){
                    if(vis[i][l]&&vis[l][j]){
                        vis[i][j]=1;
                        merge(i,j);
                    }
                }
            }
        }
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(i!=j&&par[i]==par[j]&&vis[i][j]){
                    num[i]++;
                }
            }
        }
        int ans=0;
        for(int i=1;i<=n;i++){
            if(num[i]==k) ans++;
        }
        printf("%d\n",ans);
    }
}
/*
7 2
1 2
1 3
2 4
2 5
3 6
3 7
*/

2）第二种方法：

就是保存每个点对应的边，然后dfs一遍去查找那个点对应的管理的人员有几个。然后每次判断完，然后再输出。

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
#define maxn 111
vector<int> G[maxn];
int n,res;
int vis[maxn];
void dfs(int v){
	for(int i=0;i<G[v].size();i++){
		int u=G[v][i];
		if(!vis[u]){
			res++;
			vis[u]=1;
			dfs(u);
		}
	}
}
int main(){
	int k;
	while(~scanf("%d%d",&n,&k)){
		for(int i=0;i<=maxn;i++) G[i].clear();
		memset(vis,0,sizeof(vis));
		int a,b;
		for(int i=1;i<n;i++){
			scanf("%d%d",&a,&b);
			G[a].push_back(b);
		}
		int ans=0;
		for(int i=1;i<=n;i++){
			memset(vis,0,sizeof(vis));
			res=0;
			dfs(i);
			if(res==k) ans++;
		}
		printf("%d\n",ans);
	}
}
/*
7 2
1 2
1 3
2 4
2 5
3 6
3 7
*/

//感觉第二种方法还是比较好想到的。