海岛帝国:运输资源
【试题描述】
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]);
}
}
}