这个题我做了很长时间,主要是英语没学好。
T A B : All the dragon balls which are in the same city with A have been transported to the city the Bth ball in. You can assume that the two cities are different.
意思是把A球所在城市的球全部移动到B球所在的城市。我一直以为是移动到B城市...
那么问题就简单了,总有一个球的编号和它所在城市编号相同(也就是这个集合的标记元素),而且这个球的移动次数必然为0
开一个记录每个城市的球的个数的数组ct,当且仅当球的编号i和城市的编号相同(集合的标记元素)时ct[i]有意义,表示城市i有ct[i]个球,或者编号为i的球所在的集合有ct[i]个球。合并两个集合u和v时,令ct[v] += ct[u]即可。
开一个记录每个球移动次数的数组times,对于球i而言,它上一层的编号为root[i],那么它的移动次数就是本身的移动次数加上跟着上一层球一起走的次数times[i] += times[root[i]],再对i压缩路径。
题目倒是很亲切的说T操作的两个球所在的城市不同。其实相同也很容易处理:放着不管就行了。
#include <cstdio>
int root[10001], ct[10001], times[10001];
char buf[4];
int find(int u){
if (root[u] == u) return u;
int t = find(root[u]);
times[u] += times[root[u]];
return root[u] = t;
}
inline void merge(int u, int v){
int ru = find(u), rv = find(v);
ct[rv] += ct[ru];
++times[ru];
root[ru] = rv;
}
int main(){
int T,n,q,a,b;
scanf("%d", &T);
for (int index = 1; index <= T; ++index){
printf("Case %d:\n", index);
scanf("%d %d", &n, &q);
for (int i = 1; i <= n; ++i){
root[i] = i;
ct[i] = 1;
times[i] = 0;
}
while (q--){
scanf("%s", buf);
if (buf[0] == 'T'){
scanf("%d %d", &a, &b);
merge(a,b);
}
else {
scanf("%d", &a);
int t = find(a);
printf("%d %d %d\n", t, ct[t], times[a]);
}
}
}
return 0;
}