Wireless Network POJ - 2236(并查集)

鸡山村发生了一次地震。ACM (Asia Cooperated Medical 亚洲联合医疗队) 已经为圣维尔供电中心的电脑搭建了一个无线网络,但受到了一次不可预知的余震攻击,因此网络中的所有电脑都被破坏了。电脑被逐台修复,网络逐步恢复了工作。由于受到硬件的约束,每台电脑只能与距离它不超过 d 米的其它电脑直接通信。但每台电脑可被看作其它两台电脑的通信中转点,也就是说,如果电脑 A 和电脑 B 可以直接通信,或存在一台电脑 C 既可与 A 也可与 B 通信,那么电脑 A 和电脑 B 之间就能够通信。

在处理网络修复的过程中,工作人员们在任何一个时刻,可以执行两种操作:维修一台电脑,或测试两台电脑是否能够通信。请您找出全部的测试操作。

输入

第一行包含了两个整数 N 和 d (1 <= N <= 1001, 0 <= d <= 20000)。此处 N 是电脑的数目,编号从 1 到 N;同时,D 是两台电脑之间能够直接通信的最大距离。接下来的 N 行,每行包含两个整数 xi, yi (0 <= xi, yi <= 10000),表示 N 台电脑的坐标。从第 (N+1) 行到输入结束,是逐一执行的操作,每行包含一个操作,格式是以下两者之一:
1. "O p" (1 <= p <= N),表示维护电脑 p 。
2. "S p q" (1 <= p, q <= N),表示测试电脑 p 和 q 是否能够通信。

输入不超过 300000 行。

输出

对于每个测试操作,如果两台电脑能够通信,则打印 "SUCCESS";否则,打印 "FAIL"。

示例输入

4 1
0 1
0 2
0 3
0 4
O 1
O 2
O 4
S 1 4
O 3
S 1 4

示例输出

FAIL
SUCCESS

这个题目还是很不错的,是一个并查集的变形(其中的一些用法是很厉害的,这道题目有一些要注意的点,在最后的多组输入的时候,还有这个join(并的时候)该怎么写代码,因为没有两者之间的关系输入,只是一个点一个点的坐标输入,在这里我们得需要一些技巧,来将这些点都并起来,我们才能往后判断点之间的关系)。

首先放出代码:

这道题目给出的是点的坐标,我们以前单纯的父亲数组就不行了,不能只设置一个pre[maxn]数组,因为要算两个点之间的距离,所以我们就使用结构体来代替这个pre[maxn]数组,结构体中元素是一个坐标和一个父亲点。我们开的是结构体数组,不仅能代替pre[maxn]数组,还能储存一个点的坐标。使用结构体数组在操作上是基本一样的。

在下面就是这个输入的问题了,我们该怎么写代码,来使他们每个点都判断一次是否在题目所给的距离中通信,大神在这里使用vis[maxn]数组标记每个输入进来的点,在下一次的输入中,每两个点都判断一次,如果可以在通信范围内,我们就合并集合,将所有可以通信的点都装进集合中。在判断两点是不是连同时(能不能通信),直接看看他们两点是否在一个集合内,就可以判断了。  思路还是很新颖的,大神思维新颖。

#include<iostream>
#include<cstdio>
#include<algorithm> 
const int maxn=300005;
typedef long long ll;
using namespace std;
typedef pair<int,int> p;

typedef struct
{
	int pre;
	int x,y;
}point;          //用结构体来模拟这个pre[maxn]数组 
point s[maxn];

bool vis[maxn];   //模拟每个点查和并 
int n,d;

void init()
{
	for(int i=0;i<=n;i++)
	{
		s[i].pre=i;
	}
}

int find(int x)
{
	return x==s[x].pre?x:s[x].pre=find(s[x].pre);
}

void join(point a,point b)
{
	int fx=find(a.pre);
	int fy=find(b.pre);
	if(fx!=fy)
	{
		if((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y)<=d*d)
		{
			s[fy].pre=fx;
		}
		
	}
	
}

int main()
{
	scanf("%d%d",&n,&d);
	init();
	
	for(int i=1;i<=n;i++)
	{
		scanf("%d%d",&s[i].x,&s[i].y);
	}
	
	char c;
	int num1,num2;
	while(getchar()&&scanf("%c",&c)!=EOF)
	{
		if(c=='O')
		{
			scanf("%d",&num1);
			vis[num1]=true;
			for(int i=1;i<=n;i++)
			{
				if(vis[i]&&i!=num1)
				{
					join(s[num1],s[i]);
				}
			}
		}
		else
		{
			scanf("%d%d",&num1,&num2);
			if(find(num1)==find(num2))
				printf("SUCCESS\n");
			else
				printf("FAIL\n");
		}

	}
	return 0;
}
发布了123 篇原创文章 · 获赞 83 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/tsam123/article/details/89363985