题意:
有n台损坏的电脑,现要将其逐台修复,且使其相互恢复通信功能。若两台电脑能相互通信,则有两种情况,一是他们之间的距离小于d,二是他们可以借助都可到达的第三台已修复的电脑。给出所有电脑的坐标位置,对其进行两种可能的操作,O x表示修复第x台,S x y表示判断x y之间能否通信,若能输出SUCCESS,否则输出FALL。
思路:
用并查集来保存电脑互相的连通情况。
每次修好电脑后,将它可以通信的电脑(距离满足且已修好)与它进行连通。
#include <cstdio> #include <cstring> #include <string> #include <iostream> using namespace std; const int maxn = 1000 + 5; struct node{ int x, y; }pos[maxn]; int p[maxn], vis[maxn]; int find(int x) { return p[x] == x ? x : p[x] = find(p[x]); } void unionset(int x, int y) { int px = find(x), py = find(y); if (px != py) p[px] = py; } int main() { int n, d; scanf("%d%d", &n, &d); memset(vis, 0, sizeof(vis)); for (int i = 1; i <= n; ++i) p[i] = i; for (int i = 1; i <= n; ++i) scanf("%d%d", &pos[i].x, &pos[i].y); getchar(); char s[20]; while (fgets(s, sizeof(s), stdin) != NULL) { char op; int a, b; if (s[0] == ‘S‘) { sscanf(s, "%c %d %d", &op, &a, &b); int rx = find(a), ry = find(b); if (rx == ry) printf("SUCCESS\n"); else printf("FAIL\n"); } else { sscanf(s, "%c %d", &op, &a); vis[a] = 1; int x = pos[a].x, y = pos[a].y; for (int i = 1; i <= n; ++i) { if (a == i || !vis[i]) continue; int x1 = pos[i].x, y1 = pos[i].y; if ((x - x1)*(x - x1) + (y - y1)*(y - y1) <= d*d) { unionset(a, i); } } } } return 0; }
POJ 2236 Wireless Network (并查集)
原文:http://www.cnblogs.com/demian/p/7376347.html