问题描述
就是并查集的简单应用。详情请参考
hihocoder官网。
算法详解
每个节点存储两个数值,其父亲节点的位置以及类别内部节点的个数。其中父子节点位于一个类别,而类别内部节点数只对根节点有意义。
1. 寻找根节点
根节点的位置可以用于区分两个节点的类别。所以寻找根节点是个十分常见的操作,这里使用递归实现,同时加入短路,减少平均复杂度。
int find_root(int i){
if (nodes[i].parent == -1)
return i;
return nodes[i].parent = find_root(nodes[i].parent);
}
2. 结合两个类别
题目中结合两个节点,其实也就是结合这两个节点所在的类别。鉴于每个节点只能有一个父亲节点,以保障树形结构,所以最终是需要结合两个类别对应的根节点。而为了使得树更加平衡,加入树的大小作为指标,使得较小的树作为较大树根节点的子节点。
void combine(int r1,int r2){
if (r1 == r2)
return;
if (nodes[r1].num < nodes[r2].num)
swap(r1,r2);
nodes[r2].parent = r1;
nodes[r1].num += nodes[r2].num;
nodes[r2].num = 0;
}
全部代码
#include <iostream>
#include <map>
#include <string>
using namespace std;
struct Node{
int parent;
int num;
Node():parent(-1),num(1) {}
Node(const Node &n):parent(n.parent),num(n.num) {}
~Node() {}
};
Node nodes[1000000];
int find_root(int i){
if (nodes[i].parent == -1)
return i;
return nodes[i].parent = find_root(nodes[i].parent);
}
void combine(int r1,int r2){
if (r1 == r2)
return;
if (nodes[r1].num < nodes[r2].num)
swap(r1,r2);
nodes[r2].parent = r1;
nodes[r1].num += nodes[r2].num;
nodes[r2].num = 0;
}
void print_nodes(int cnt){
cout << "parent:\t";
for (int i = 0;i < cnt;++i)
cout << nodes[i].parent << '\t';
cout << endl;
cout << "size:\t";
for (int i = 0;i < cnt;++i)
cout << nodes[i].num << '\t';
cout << endl;
}
map<string,int> str2int;
int main()
{
int N,cnt=0;
cin >> N;
int op;
string str1,str2;
for (int i = 0;i < N;++i){
cin >> op >> str1 >> str2;
if (str2int.find(str1) == str2int.end())
str2int.insert(pair<string,int>(str1,cnt++));
if (str2int.find(str2) == str2int.end())
str2int.insert(pair<string,int>(str2,cnt++));
if (op == 0)
combine(find_root(str2int[str1]),find_root(str2int[str2]));
else
cout << ((find_root(str2int[str1])==find_root(str2int[str2]))?"yes":"no") << endl;
}
return 0;
}