并查集练习-帅到没朋友

小练习记录贴06

点击跳转到原题

---

题目描述

当芸芸众生忙着在朋友圈中发照片的时候，总有一些人因为太帅而没有朋友。本题就要求你找出那些帅到没有朋友的人。

输入格式

输入第一行给出一个正整数N（≤100），是已知朋友圈的个数；随后N行，每行首先给出一个正整数K（≤1000），为朋友圈中的人数，然后列出一个朋友圈内的所有人——为方便起见，每人对应一个ID号，为5位数字（从00000到99999），ID间以空格分隔；之后给出一个正整数M（≤10000），为待查询的人数；随后一行中列出M个待查询的ID，以空格分隔。
注意：没有朋友的人可以是根本没安装“朋友圈”，也可以是只有自己一个人在朋友圈的人。虽然有个别自恋狂会自己把自己反复加进朋友圈，但题目保证所有K超过1的朋友圈里都至少有2个不同的人。

输出格式

按输入的顺序输出那些帅到没朋友的人。ID间用1个空格分隔，行的首尾不得有多余空格。如果没有人太帅，则输出No one is handsome。
注意：同一个人可以被查询多次，但只输出一次。

输入样例

样例1

3
3 11111 22222 55555
2 33333 44444
4 55555 66666 99999 77777
8
55555 44444 10000 88888 22222 11111 23333 88888

样例2

3
3 11111 22222 55555
2 33333 44444
4 55555 66666 99999 77777
4
55555 44444 22222 11111

输出样例

样例1

10000 88888 23333

样例2

No one is handsome

---

解题思路

找人物关系，可以用并查集来解决。用一个root数组记录当前i的root是什么，一开始的时候当然都是自己。用一个sum数组记录当前这个root下有多少个节点。最后判断的时候，如果root是本身并且sum==1，那么说明ta的朋友圈里除了自己没有任何人。当然还需要一个map来判断一下这个数组之前有没有判断过，开一个bool数组记录也是可以的，但是map不是更方便吗？

需要注意的是，输入的id都是五位数的，结果也要输出五位数的。也就说，1要以00001形式出现。这个地方坑到我了。

找到了一年前因为上述原因没有ac的代码，修改了输出05d也过了。发现只需要记录有没有出现过就可以了。如果当前k==1，那么跳过。

Talk is cheap, show me the code

version1

#include<iostream>
#include<map>
using namespace std;
const int N = 1e5 + 5;
int root[N], sum[N];

int find(int num){
	return root[num] == num ? num : root[num] = find(root[num]);
}


void union_(int a, int b){
	int rootA = find(a);
	int rootB = find(b);
	if(rootA != rootB){
		root[a] = rootB;
		sum[rootB] += sum[rootA];
	}
}

int main(){
	int n; cin >> n;
	for(int i = 0; i < N; i ++) root[i] = i, sum[i] = 1;
	while(n --){
		int k; cin >> k;
		int r; 
		for(int i = 0; i < k; i ++){
			if(!i) cin >> r;
			else{
				int num; cin >> num;
				if(num != r) union_(r, num);
			}
		}
	}
	int m; cin >> m;
	map<int, bool> book;
	bool flag = true;
	while(m --){
		int num; cin >> num;
		if(!book[num]){
			book[num] = true;
			if(root[num] == num && sum[num] == 1){
				if(!flag) cout << " ";
				cout << num;
				if(flag) flag = false;
			}
		}
	}
	if(flag) cout << "No one is handsome";
	return 0;
}

version2

#include<iostream>
#include<vector>
using namespace std;
bool book[100005];
bool book2[100005];
int main(){
    int N;cin>>N;
    while(N--){
        int K;cin>>K;
        if(K>1){
            for(int i=0;i<K;i++){
                int num;cin>>num;
                book[num]=true;
            }
        }else{
            int num;cin>>num;
        }
    }
    int M;cin>>M;
    vector<int> v;
    while(M--){
        int num;cin>>num;
        if(book[num]==false){
            if(book2[num]==false){
                v.push_back(num);
                book2[num]=true;
            }
        }
    }
    if(v.empty()){cout<<"No one is handsome";}
    else{for(int i=0;i<v.size();i++){printf("%05d",v[i]);if(i<v.size()-1) cout<<" ";}}
    return 0;
}