PAT甲级——并查集专题

1107 Social Clusters (30 分)

https://pintia.cn/problem-sets/994805342720868352/problems/994805361586847744

题意：n个人，每个人有k个爱好，有相同爱好的人是一个社交集合，问有几个社交集合及他们各自的人数。 

思路：每个人都父节点初值赋为自己，随着输入将该节点与有同样爱好的的根节点合并，最后遍历每个该节点的根节点++，

统计不为0的isRoot，排序输出；

注意：find函数不仅找到根节点，而且将“沿途”结点的pre都赋值为最终根节点。

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
vector<int>pre,isRoot;
int hobby[1001];
int find(int x){
	int y=x;
	while(x!=pre[x]){  //找到x的根 
		x=pre[x];
	}
	while(y!=pre[y]){  //将路径上的点的根节点都设为最终的根节点 
		int tmp=y;     //先预存结点值，结点值后移一位，原结点的根节点赋值为最终根节点 
		y=pre[y];
		pre[tmp]=x;
	}
	return x;
}
void merge(int x,int y){
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
		pre[fx]=fy;
}
bool cmp(int a,int b){
	return a>b;
}
int main(){
	//ios::sync_with_stdio(false);
	int n,k,x;
	cin>>n;
	pre.resize(n+1);
	isRoot.resize(n+1);
	for(int i=1;i<=n;i++) //pre数组赋初值 
	 	pre[i]=i;
	for(int i=1;i<=n;i++){
		scanf("%d:",&k);
		for(int j=0;j<k;j++){
			cin>>x;
			if(hobby[x]==0) hobby[x]=i; //存根节点 
			merge(i,find(hobby[x])); //合并组 
		}
	}
	int cnt=0;
	for(int i=1;i<=n;i++) //记录某点作为根的数量 
		isRoot[find(i)]++;
	for(int i=1;i<=n;i++) //记录组的数目 
		if(isRoot[i]!=0) cnt++;
	cout<<cnt<<endl;
	sort(isRoot.begin(),isRoot.end(),cmp);
	for(int i=0;i<cnt;i++){
		if(i!=0) cout<<" ";
		cout<<isRoot[i];
	}
	return 0;
} 

1118 Birds in Forest (25 分)

https://pintia.cn/problem-sets/994805342720868352/problems/994805354108403712

题意：拍了n张照片，每张照片上有k个鸟，每只鸟只可能在一只树上，鸟的编号为连续的从1到某一数字，要求输出一共有多少棵树，一共有多少只鸟。此外还有共q次询问，每次输入两只鸟的编号，若两只鸟在一棵树上输出Yes，反之No。

思路：因为是连续编号，所以在输入过程中遇到的最大节点就是鸟的数量。tree【i】用来放第i棵树的第一只鸟，每张照片上的鸟和该树第一只鸟merge，最后计数的时候遍历鸟的个数，将该节点的根节点++，统计isRoot不为零的结点即为树的个数。

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
vector<int>pre,isRoot;
int tree[10001];
int find(int x){
	int y=x;
	while(x!=pre[x]){  //找到x的根 
		x=pre[x];
	}
	while(y!=pre[y]){  //将路径上的点的根节点都设为最终的根节点 
		int tmp=y;     //先预存结点值，结点值后移一位，原结点的根节点赋值为最终根节点 
		y=pre[y];
		pre[tmp]=x;
	}
	return x;
}
void merge(int x,int y){
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
		pre[fx]=fy;
}
bool cmp(int a,int b){
	return a>b;
}
int main(){
	//ios::sync_with_stdio(false);
	int n,k,x,num=0;
	cin>>n;
	pre.resize(10001);
	isRoot.resize(10001);
	for(int i=1;i<=10000;i++) //pre数组赋初值 
	 	pre[i]=i;
	for(int i=1;i<=n;i++){
		cin>>k;
		for(int j=0;j<k;j++){
			cin>>x;
			num=max(num,x);
			if(tree[i]==0) tree[i]=x; //存根节点 
			merge(x,find(tree[i])); //合并组 
		}
	}
	int cnt=0,q,a,b;
	for(int i=1;i<=num;i++) //记录某点作为根的数量 
		isRoot[find(i)]++;
	for(int i=1;i<=num;i++) //记录组的数目 
		if(isRoot[i]!=0) cnt++;
	cout<<cnt<<" "<<num<<endl;
	cin>>q;
	for(int i=0;i<q;i++){
		cin>>a>>b;
		if(find(a)==find(b))
			cout<<"Yes\n";
		else cout<<"No\n";
	}
	return 0;
} 

 或者更一般的情况（即使鸟的编号不连续也适用），用一数组标记是否该节点是一只鸟的编号，遍历1到10000，若该节点true，将该节点的根节点++，统计该节点存在且cnt不为0的个数即为树的个数；鸟的数量为cnt【i】的累加和。

 

1114 Family Property (25 分)

https://pintia.cn/problem-sets/994805342720868352/problems/994805356599820288

题意：给出n个id，及他们的父母id（父母去世id为-1），孩子个数，孩子们的id，拥有房产数目及房产面积数。求一共有多少个大家庭（所有有亲属关系的人都属于同一个家庭），按房产面积递减、最小编号递增的顺序输出每个大家庭的最小编号、人数、平均房产套数，平均房产面积数。

思路：

• 两个结构体，data存数据（0到n-1），ans放输出信息（0000-9999，以flag标识）。
• 存入过程中顺便merge和vis标记。
• 遍历data（0到n-1)， 累加data结点的根节点的ans的房产数和面积数，（ans用结构体里的flag标记）。
• 遍历ans（0000到9999），通过vis统计点 i 的根节点的ans的people，通过flag统计家庭个数cnt。
• 遍历ans（0000到9999），通过flag求各家庭均值。
• sort，通过flag输出。

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
struct Data {
	int id,fid,mid;
	int cid[6];
	double cnt,area;
} data[10005];
struct node {
	int id,people;
	double cnt,area;
	bool flag=false;
} ans[10005];
int pre[10005];
bool vis[10005];
int find(int x){
	int y=x;
	while(x!=pre[x]){
		x=pre[x];
	}
	while(y!=pre[y]){
		int tmp=y;
		y=pre[y];
		pre[tmp]=x;
	}
	return x;
}
void merge(int a,int b){
	int fa=find(a);
	int fb=find(b);
	if(fa>fb) pre[fa]=fb;
	else if(fa<fb) pre[fb]=fa;
}
bool cmp(node a,node b){
	if(a.area!=b.area) return a.area>b.area;
	else return a.id<b.id; 
}
int main() {
	int n,id,k;
	cin>>n;
	for(int i=0;i<10000;i++) pre[i]=i;
	for(int i=0; i<n; i++) {
		cin>>data[i].id>>data[i].fid>>data[i].mid>>k;
		vis[data[i].id]=true;
		if(data[i].fid!=-1){
			vis[data[i].fid]=true;
			merge(data[i].id,data[i].fid);
		}
		if(data[i].mid!=-1){
			vis[data[i].mid]=true;
			merge(data[i].id,data[i].mid);
		}
		for(int j=0; j<k; j++) {
			cin>>data[i].cid[j];
			vis[data[i].cid[j]]=true;
			merge(data[i].cid[j],data[i].id);
		}
		cin>>data[i].cnt>>data[i].area;
	}
	for(int i=0; i<n; i++) { //房产什么的n套
		id=find(data[i].id);
		ans[id].id=id;
		ans[id].cnt+=data[i].cnt;
		ans[id].area+=data[i].area;
		ans[id].flag=true;
	}
	int cnt=0; //家庭个数
	for(int i=0; i<10000; i++) { //统计家里一共多少人，遍历
		if(vis[i]) ans[find(i)].people++;
		if(ans[i].flag) cnt++;
	}
	for(int i=0; i<10000; i++) {
		if(ans[i].flag) {
			ans[i].cnt=ans[i].cnt/ans[i].people;
			ans[i].area=ans[i].area/ans[i].people;
		}
	}
	sort(ans,ans+10000,cmp);
	cout<<cnt<<endl;
	for(int i=0; i<cnt; i++) {
		if(ans[i].flag) {
			printf("%04d %d %.3lf %.3lf\n",ans[i].id,ans[i].people,ans[i].cnt,ans[i].area);
		}
	}
	return 0;
}