题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=3635
题目大意:
初始时,有n个龙珠,编号从1到n,分别对应的放在编号从1到n的城市中。
现在又2种操作:
T A B,表示把A球所在城市全部的龙珠全部转移到B城市。(第一次时,因为A球所在的城市只有一个球,所以只移动1个,如果有多个,则全部移动)。
Q A,表示查询A。要求得到的信息分别是:A现在所在的城市,A所在城市的龙珠数目,A转移到该城市移动的次数(如果没有移动就输出0)
解题思路:
并查集的应用。但是这道题让我纠结了很久,首先是转移次数怎么确定,经过思考,我们可以开一个辅助数组,当转移一次时,我们就把要转移的城市的全部龙珠的转移次数全部做+1操作。但是遇到的问题就是超时问题,我们需要遍历全部龙珠,找出要转移的城市的全部龙珠然后才能够+1,这样,很容易就超时了。
反复想了很久,想了另一个办法,就是通过路径压缩来更新转移的次数,比如每次移动时,我只需要把这个城市的根结点的转移次数+1,等到以后路径压缩时,子结点自己移动的次数加上根结点移动的次数,就是这个结点总共的移动次数,不明白的可以自己动手画画。。
另一个蛋疼的问题就是printf。。。上面那个问题纠结了10几分钟就搞定了,这个问题一下子搞了3个小时,快悲剧死。。。。。。。
不过也算不错,知道printf是倒序输出的,它默认的输出是从右到左,当最右边的变量受前面变量的影响时,答案肯定错误,因为它先计算的是最右边的值。。。好吧,我承认自己C语言学的还不行,要不然也不会在这里翻船了。。。看来一些简单的东西也是值得研究的。。。ORZ!~
通过这道题,对并查集的查找又有了一定的认识,初步理解了路径压缩的好处,这里,根结点的转移次数不是直接加上根结点的移动次数,而是在路径压缩的过程中逐层累加,比如有4层,根结点为1,然后第二层2,第三层3,第四层4.那么,第一层路径压缩过程中进入第二层,然后第三层,然后第四层,第四层也就是根,找到根结点,返回第三层,第三层+上第四层的转移次数,然后返回第二层,第二层转移次数加上第三层转移次数,返回第一层,第一层转移次数加上第二层转移次数,同时,在这个过程中也把下面3层的结点的父节点直接更新为和根结点,也就是1相连,查找结束后,称为2层结构,根结点1在一层,234都处于第二层。这样的一个过程是通过递归调用本身完成的,跟数据结构(严蔚敏版本)讲解栈那节的汉诺塔过程相似,如果想完全掌握,推荐仔细看教材+自己模拟,这样就可以理解了。。。。。
代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 10010
struct node
{
int parent; //根节点
int son; //子节点
int transport; //转移次数
}p[N];
int find(int x)
{
int temp;
if(x == p[x].parent)
return x;
else
{
temp = p[x].parent;
p[x].parent = find(p[x].parent);
p[x].transport += p[temp].transport; //这点需要反复揣摩
}
return p[x].parent;
}
void join(int x, int y)
{
int root1, root2;
root1 = find(x);
root2 = find(y);
if(root1 == root2)
return ;
p[root1].parent = root2;
p[root1].transport++; //根节点转移次数+1,要理解
p[root2].son += p[root1].son;
p[root1].son = 0;
}
int main()
{
int ncase, T = 1;
int num, querynum;
char ope;
int from, to;
int querycity;
int ans;
scanf("%d", &ncase);
while(ncase--)
{
scanf("%d%d", &num, &querynum);
for(int i = 1; i <= num; ++i) //初始化
{
p[i].parent = i;
p[i].son = 1;
p[i].transport = 0;
}
printf("Case %d:\n", T++);
for(int i = 0; i < querynum; ++i)
{
scanf("%*c%c", &ope);
if(ope == 'T')
{
scanf("%d%d", &from, &to);
join(from, to);
}
else
{
scanf("%d", &querycity);
ans = find(querycity);
printf("%d %d %d\n", ans, p[ans].son, p[querycity].transport); //纠结了3个小时~~
}
}
}
return 0;
}