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

Posted 2020-07-28 wxjor

海岛帝国：运输资源

【试题描述】

        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。具体代码如下：

1 void Union(int a,int b)
2 {
3     int pa=find(a);
4     int pb=find(b);
5     par[pa]=pb;
6     size[pb]+=size[pa];
7     step[pa]++;
8 }

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

 1 void init(int n)
 2 {
 3     for(int i=1;i<=n;i++)
 4     {
 5         par[i]=i;
 6         step[i]=0;
 7         size[i]=1;
 8     }
 9 }
10  

【代码如下】

 1 #include <cstdio>
 2 #include <cstring>
 3 using namespace std;
 4 #define MAX 10
 5 int par[MAX],step[MAX],size[MAX];
 6 void init(int n)
 7 {
 8     for(int i=1;i<=n;i++)
 9     {
10         par[i]=i;
11         step[i]=0;
12         size[i]=1;
13     }
14 }
15 
16 int find(int x)
17 {
18     if(x==par[x])    return x;
19     int tmp=par[x];
20     par[x]=find(tmp);
21     step[x]+=step[tmp];
22     return par[x];
23 } 
24 
25 void Union(int a,int b)
26 {
27     int pa=find(a);
28     int pb=find(b);
29     par[pa]=pb;
30     size[pb]+=size[pa];
31     step[pa]++;
32 }
33 
34 int main()
35 {
36     int T,n,m,a,b,t=1;
37         scanf("%d%d",&n,&m);
38         init(n);
39         for(int i=0;i<m;i++)
40         {
41             char move;
42             getchar();
43             move=getchar();
44             if(move==\'T\')
45             {
46                 scanf("%d%d",&a,&b);
47                 Union(a,b);
48             }
49             else
50             {
51                 scanf("%d",&a);
52                 int pa=find(a);
53                 printf("%d %d %d\\n",pa,size[pa],step[a]);
54             }
55         }
56 }

View Code