并查集
并查集就是把节点放在数组List里面,初始化让List里面的每个位置都等于该位置角标。之后填入数据(用Union),Union就是先找根,如果根不一样,就把aRoot并到bRoot底下。完成填入操作后,如果List里面的某个i还是角标本身,就说明这个点是根节点。
PTA 1107 Social Clusters
大意是给了N个人,每个人有一些爱好,如果有相同爱好就是好朋友。这样的话就会形成一个好朋友社交圈。为你最后有几个这样的大圈。
解:
用一个course数组储存爱好,一开始都是0,当第一次出现某个人有这个爱好的时候,就把course数组里面对应的爱好记成这个人的编号。这样下次出现这个爱好的时候,就把下次出现的人的编号和已经储存的编号相连。 这样就完成了List填入。
之后都是常规操作了。
核心还是要会写findRoot以及Union函数。
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
const int maxnum = 1005;
const int hobbyNum = 1001;
int course[hobbyNum] = {
0 };
int list[maxnum];
void Intil()
{
for (int i = 0; i < maxnum; i++)
list[i] = i;
}
int findRoot(int a) //找某节点的根
{
while (a != list[a])
a = list[a];
return a;
}
void Union(int a, int b)
{
int aRoot = findRoot(a);
int bRoot = findRoot(b);
if (aRoot != bRoot) //根不一样,合并
list[aRoot] = bRoot;
}
bool cmp(int a, int b)
{
return a > b ? true : false;
}
int main()
{
int N;
cin >> N;
Intil(); //初始化人头数组
for