题目链接: POJ 2236 Wireless Network
题目大意:
有N台的计算机,要用他们组成一个无线网络。 两台计算机之间的距离小于D则可以直接相连,大于D,则需要有中间节点来间接连接这两台计算机。给出所有计算机的坐标,初始状态所有的计算机都是坏的, 需要进行修复。
工程师可以进行两种操作,修复一台计算机或者测试两台计算机的连通性
输入”O x”表示修复第i台计算机
输入”S x y”表示测试计算机x, y的连通性, 连通则输出”SUCCESS”, 不连通则输出”FAIL”
输入格式:
第一行 N, D
接下来N行, 每行两个数, 表示第i台计算机的坐标
接下来一直到输入结束,每行一个操作, 或者表示修复,或者表示测试
输出格式:
对于每个测试操作,连通则输出”SUCCESS”, 不连通则输出”FAIL”
题解:
典型的并查集操作。由于距离的限制, 每次修复一台计算机,只需要检查可以和它直接相连的计算机是否有已经被修复过的, 然后Union这两个集合即可。基于这个想法, 我在进行修复操作之前,先给这些电脑建了个图。为此付出了 n2 的代价。对于这题来说,或许有些多此一举了,说不定反而浪费了时间。毕竟N最大只有1001。完全可以直接枚举所有的其他计算机。但我觉得当n再增大个十几倍或许建图就十分必要了。
另外, c的字符输入要谨慎。不小心就会超时。EOF还是要自己判断一下的。
代码:
#include <iostream>
#include <vector>
#include <cstring>
#include <cstdio>
#define MAXN 1010
using namespace std;
pair<int, int> computer[MAXN]; // 计算机的坐标
int N, D;
int f[MAXN]; // 并查集的父节点
bool repaired[MAXN]; // 是否修复过
vector<int> *adj; //图的邻接表表示
//是否相邻
bool is_adj(int a, int b) {
int dx = computer[a].first - computer[b].first;
int dy = computer[a].second - computer[b].second;
return dx*dx + dy*dy <= D*D;
}
//建图
void buildGraph() {
adj = new vector<int>[N+1];
for (int i = 1; i <= N-1; i++) {
for (int j = i+1; j <= N; j++) {
if (is_adj(i, j)) {
adj[i].push_back(j);
adj[j].push_back(i);
}
}
}
}
int Find(int x) {
if (f[x] == x) return x;
else return f[x] = Find(f[x]);
}
void Union(int x, int y) {
int fx = Find(x);
int fy = Find(y);
if (fx != fy) {
f[fx] = fy;
}
}
int main() {
//输入
scanf("%d%d", &N, &D);
for (int i = 1; i <= N; i++)
scanf("%d%d", &computer[i].first, &computer[i].second);
//初始化
memset(repaired, false, sizeof(repaired));
for (int i = 1; i <= N; i++) f[i] = i;
buildGraph(); //建图
//操作
int operation;
int x, y;
while ((operation = getchar()) != EOF) { // AC
//对比其他输入方式
//while (scanf("%c", &operation)) { // Time Limit Exceeded
//while (cin >> operation) { // AC
if (operation)
if (operation == 'O') {
scanf("%d", &x);
repaired[x] = true;
vector<int>::iterator i;
for (i = adj[x].begin(); i != adj[x].end(); i++) {
if (repaired[*i]) Union(x, *i);
}
}
else if (operation == 'S') {
scanf("%d%d", &x, &y);
if (Find(x) == Find(y)) printf("SUCCESS\n");
else printf("FAIL\n");
}
}
return 0;
}