杭电3172——并差集(map容器)
写在题前:
并差集的简单应用吧,无非就是用到map容器,以前只给节点,我们可以直接当成数字,现在节点换成字符串了,用map容器转换一下就行。
解题思路:
- 使用map容器,让每一个字符串都对应唯一一个数字;
- 用num数组记录每一区域的个数。
写在题后:
这道题基本没坑,注意两点:
- 就是输入的时候,嵌套有点多,注意数据重置。
- 还有就是如果一个一个遍历的话,会超时的(要用num数组存)。
ac代码
# include <iostream>
# include <algorithm>
# include <cstdio>
# include <cstdlib>
# include <string>
# include <cstring>
# include <vector>
# include <map>
using namespace std;
const int maxn = 2e5 + 10;
int id[maxn];//记录父亲节点
int num[maxn];//记录节点所在区域的个数。
map<string, int>m;
int fin(int x) {//路径压缩
return id[x] = id[x] == x ? x : fin(id[x]);
}
int meg(int x, int y) {//连接x和y
int a = fin(x);
int b = fin(y);
if (a != b) {
id[b] = a;
num[a] += num[b];//两个区域建立了连接,num数组要更新。
}
return num[a];
}
int main() {
ios::sync_with_stdio(false);
int T;
while (cin >> T) {
while (T--) {
for (int i = 0; i < maxn; i++) {
id[i] = i;
num[i] = 1;//初始值为1,。
}
m.clear();
int n;
cin >> n;
for (int i = 0; i < n; i++) {
string t1;
string t2;
cin >> t1 >> t2;
m.insert(pair<string, int>(t1, m.size()));
m.insert(pair<string, int>(t2, m.size()));
int ans = meg(m[t1], m[t2]);//用m[t]访问字符串t的数字。
cout << ans << endl;
}
}
}
return 0;
}