POJ-2236 Wireless Network （簡單並查集）

也是幾個月前做的題目，填坑

POJ-2236 Wireless Network

Description

南亞發生了一次地震。ACM (Asia Cooperated Medical 亞洲聯合醫療隊) 已經爲膝上型電腦搭建了一個無線網絡，但受到了一次不可預知的餘震攻擊，因此網絡中的所有電腦都被破壞了。電腦被逐臺修復，網絡逐步恢復了工作。由於受到硬件的約束，每臺電腦只能與距離它不超過 d 米的其它電腦直接通信。但每臺電腦可被看作其它兩臺電腦的通信中轉點，也就是說，如果電腦 A 和電腦 B 可以直接通信，或存在一臺電腦 C 既可與 A 也可與 B 通信，那麼電腦 A 和電腦 B 之間就能夠通信。

在處理網絡修復的過程中，工作人員們在任何一個時刻，可以執行兩種操作：維修一臺電腦，或測試兩臺電腦是否能夠通信。請您找出全部的測試操作。

Input

第一行包含了兩個整數 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 行。

Output

對於每個測試操作，如果兩臺電腦能夠通信，則打印 “SUCCESS”；否則，打印 “FAIL”。

Sample Input

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

Sample Output

FAIL
SUCCESS

AC代碼：

#include<bits/stdc++.h>
#define maxn 1010
using namespace std;

struct com{
   int x,y;
   bool isok;
};

struct com a[maxn];
int pre[maxn];
double dist[maxn][maxn];


int find_r(int x)
{
    if(pre[x]==x) return x;
    else{
        int root = find_r(pre[x]);
        pre[x] = root;
        return root;
    }//尋根同時化簡併查集
}

void unite(int x, int y)
{
    x = find_r(x); y = find_r(y);
    pre[x] = y;
    return;
}//合併兩個並查集

int N,d;

int main()
{
    scanf("%d%d",&N,&d);
    for(int i = 1; i <= N; i++)
    {
        scanf("%d %d",&a[i].x, &a[i].y);
        pre[i] = i; a[i].isok = false;
    }
    
    // 計算各電腦之間的距離
    for(int i = 1; i <= N; i++)
        for(int j = i; j <= N; j++)
    {
        if(i!=j)
            dist [j][i] = dist[i][j] = sqrt((a[i].x - a[j].x)*(a[i].x - a[j].x) + (a[i].y - a[j].y)*(a[i].y - a[j].y));
        else dist[i][j] = 0;
    }
    
    char op;
    getchar();
    while(scanf("%c",&op)!=EOF)
    {
        if(op=='O'){
            int k;
            scanf("%d",&k);
            a[k].isok = true;
            for(int i = 0; i <= N; i++)
                if(dist[k][i] <= d && a[i].isok && i!=k) // 一定要兩臺電腦都正常工作才能合併
                    unite(k,i);
        }
        else {
            int x,y;
            scanf("%d %d",&x, &y);
            x = find_r(x); y = find_r(y);
            if(x==y && a[x].isok && a[y].isok) puts("SUCCESS"); // 一定要兩臺電腦都正常工作才能合併
            else puts("FAIL");
        }
        getchar();
    }

    return 0;
}