【海岛帝国系列赛】No.3 海岛帝国：运输资源

海岛帝国：运输资源

【试题描述】

YSF考虑到“药师傅”帝国现在资源极度不平均，于是，商讨启用南水北调工程。YZM为首席工程师。现在，YSF由于工作紧张，准备军用物资和民用物资。但他要时时关注运输工程动态，因为货物很珍贵，是最新发明的T-2500智能生产行业制造机。用于工、农业生产，而且技术先进，数量众多，不能落入恐怖分子之手。YSF要YZM向他随时汇报货物走向、货物地点，以防万一恐怖分子袭击。脾气暴躁的YZM一向不爱干这种统筹规划的杂活。所以，YZM需要你来帮他编一个小程序，来随时给多疑的YSF汇报。（程序当然是有薪水的，至于多少嘛……可以找YSF,LJX面谈）

【输入要求】

* 第一行：两个数n，m，表示一共有n个制造机，m条操作。操作有两种。

* 接下来的m行：T a b 表示把a制造机所在的城里的所有龙珠运到b所在的城里

Q a 表示对a的询问，要求输出x a所在的城， y a所在的城里一共有多少个机器，z a经过几次到达现在所在的城的。

【输出要求】

* 及每次Q查询的答案，回车！回车！回车！

【输入实例】

3 4
Q 1
T 3 2
Q 2
Q 3

【输出实例】

1 1 0
2 2 0
2 2 1

【其他说明】

所有数据不大于40

【试题分析】

第一眼上来就看到合并，查询等一些操作字符了，第一反应貌似是线段树？可惜，如果一上来这么分析，而不考虑麻烦程度的话，那我也就没话可说了。线段树操作这题貌似是能解决，但是本人不怎么会线段树，所以我就又蒙了个模拟线段树的并查集算法？呵呵呵。因为我想都是树形结构。所以就套了一发，其实本来就是一样的啊。定义parent同并查集的一般操作，step表示经过几步到达现在所在的城，size表示该城里的龙珠数。上回讲解并查集时提到了其它多种操作：Union。具体代码如下：

 void Union(int a,int b)
 {
     int pa=find(a);
     int pb=find(b);
     par[pa]=pb;
     size[pb]+=size[pa];
     step[pa]++;
 }

当然，定义初值要按题意定义：

 void init(int n)
 {
     for(int i=;i<=n;i++)
     {
         par[i]=i;
         step[i]=;
         size[i]=;
     }
 }
  

【代码如下】

 #include <cstdio>
 #include <cstring>
 using namespace std;
 #define MAX 10
 int par[MAX],step[MAX],size[MAX];
 void init(int n)
 {
     for(int i=;i<=n;i++)
     {
         par[i]=i;
         step[i]=;
         size[i]=;
     }
 }

 int find(int x)
 {
     if(x==par[x])    return x;
     int tmp=par[x];
     par[x]=find(tmp);
     step[x]+=step[tmp];
     return par[x];
 } 

 void Union(int a,int b)
 {
     int pa=find(a);
     int pb=find(b);
     par[pa]=pb;
     size[pb]+=size[pa];
     step[pa]++;
 }

 int main()
 {
     int T,n,m,a,b,t=;
         scanf("%d%d",&n,&m);
         init(n);
         for(int i=;i<m;i++)
         {
             char move;
             getchar();
             move=getchar();
             if(move=='T')
             {
                 scanf("%d%d",&a,&b);
                 Union(a,b);
             }
             else
             {
                 scanf("%d",&a);
                 int pa=find(a);
                 printf("%d %d %d\n",pa,size[pa],step[a]);
             }
         }
 }