第一行 电脑数量n, 最小联络距离d
以后n行 顺序给出电脑坐标
之后输入:首字母为 'O' ,然后给出修理的电脑编号i,电脑 i 可以使用
首字母为 'S',然后测试电脑 i j 之间能否通信
通信要求: 电脑完好,且电脑之间距离小于d,可以通过其他电脑间接联系
思路: 并查集 电脑修好之后遍历其他电脑查看是否可以合并
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <queue>
#include <cmath>
#include <fstream>
const int MAX = 1e4+100;
const int INF = 1e6;
using namespace std;
int n;
double d;
char opt;
int par[MAX];//根节点
int ran[MAX];//树深
struct cmu_{
double x, y;
bool u;
};
cmu_ pc[MAX];
//初始
void Init(int n){
for(int i=1; i<=n; i++){
par[i] = i;
ran[i] = 0;
}
}
//查询
int Find(int x){
if(x == par[x])
return par[x];
else
return par[x] = Find(par[x]);
}
//合并
void Unit(int a, int b){
a = Find(a);
b = Find(b);
if(a == b)
return ;
if(ran[a] < ran[b])
par[a] = b;
else{
par[b] = a;
if(ran[a] == ran[b])
ran[a]++;
}
}
//判断
bool Same(int a, int b){
return Find(a) == Find(b);
}
//测试函数
int main(){
ifstream cin ("D:\\钢铁程序员\\程序数据\\059无线网络.txt");//从文件读取数据流,省去手动输入的麻烦
if(!cin){//读取如果失败
cout << "ERROR" << endl;
}
cin >> n >> d;
for(int i=1; i<=n; i++){
cin >> pc[i].x >> pc[i].y;
pc[i].u = false;
}
Init(n);
while(cin >> opt){
int a, b;
//要修理 修理完毕尝试合并周围pc
if(opt == 'O'){
cin >> a;
pc[a].u = true;
//尝试合并
for(int i=1; i<=n; i++){
if(i == a)continue;
if(pc[i].u){//已修好 且 距离小于d
double D = (pc[a].x-pc[i].x)*(pc[a].x-pc[i].x) + (pc[i].y-pc[a].y)*(pc[i].y-pc[a].y);
if(D <= d*d)
Unit(a, i);
}
}
}
//要检查
else{
cin >> a >> b;
if(Same(a, b))
cout << "SUCCESS" << endl;
else
cout << "FAIL" << endl;
}
}
cin.close();//打开文件以后要关闭
return 0;
}