刷题向 | 1004 Counting Leaves

输入：

n、m（全部结点个数、非叶结点个数）

id（父结点id）  j（子节点个数）   id1、id2……idj

……（m行）

输出：

所有层数叶子结点个数

 

在合并时，是两个点的源结点相连，则每次合并源结点层数c肯定都是0，并且查询时，会将所有源点下的所有子结点直接指向源结点，当某个源结点成为另一棵树的子树时，变为层数为c0的点时，此点下的其他点层数的更新只需要加上c0（好吧，我也不知道自己在说什么了，很奇怪的做法）

#include<stdio.h>
#include<vector>
#include<algorithm>
#include<string.h>
#include<iostream>
#include<limits>
#include<fstream>
#include<math.h>
#include<stack>
#include<queue>
#include<bitset>
#include<utility>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
const int N=105;
const ll mod=1e9+7;
const int M=10000;
int n,m;
int pre[N];
int c[N];
int ans[N];
bool f[N];
int find(int x){
	if(pre[x]==x)return x;
	int r=find(pre[x]);	
	if(pre[x]!=r){			
		c[x]+=c[pre[x]];
		pre[x]=r;
	}
	return r;
}
void add(int x,int y){
	int rx=find(x);
	int ry=find(y);
	if(rx!=ry){
		pre[ry]=rx;
		c[y]=c[x]+1;
	}
}
int main(){
	while(cin>>n){
		if(n==0)break;
		cin>>m;
		for(int i=0;i<n;i++){
			pre[i]=i;
		}
		memset(c,0,sizeof(c));
		memset(f,0,sizeof(f));
		map<string,int> mp;
		map<string,int>::iterator iter;
		int j;
		string s;
		int pr,nr;
		while(m--){
			cin>>s;
			iter=mp.find(s);
			if(iter!=mp.end()){
				pr=iter->second;
			}
			else{
				pr=mp.size();
				mp.insert(make_pair(s,pr));
			}
			f[pr]=1;
			cin>>j;
			while(j--){
				cin>>s;
				iter=mp.find(s);
				if(iter!=mp.end()){
					nr=iter->second;
				}
				else{
					nr=mp.size();
					mp.insert(make_pair(s,nr));
				}
				add(pr,nr);
			}
		}
		memset(ans,0,sizeof(ans));
		int mc=0;
		for(int i=0;i<n;i++){
			find(i);
			mc=max(mc,c[i]);
			if(f[i]==0){
				ans[c[i]]++;
			}			
		}
		for(int i=0;i<=mc;i++){
			cout<<ans[i];
			if(i==mc)cout<<endl;
			else cout<<" ";
		}
	}
	
    return 0;
}