并查集就好了,先预处理一下,查看每一个到另外一个是否是直接可到达的,并存于数组中。
然后每一次询问,都查找一下两个是否在一个集合中。
如果是输入的O,就添加进入可以到达的集合。
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<climits>
#include<stack>
#include<vector>
#include<queue>
#define up(i,a,b) for(int i=a;i<b;i++)
#define dw(i,a,b) for(int i=a;i>b;i--)
#define upd(i,a,b) for(int i=a;i<=b;i++)
#define dwd(i,a,b) for(int i=a;i>=b;i--)
//#define local
typedef long long ll;
const double esp = 1e-6;
const double pi = acos(-1.0);
const long long INF = 0x3f3f3f3f;
using namespace std;
int n, d;
int ht[1005], fr[1005];
int use[1005], ok[1005][1005];
struct a
{
int x, y;
}cmp[1005];
int distance(int x1, int y1, int x2, int y2)//距离判定
{
if (((x1 - x2)*(x1 - x2) + (y1 - y2)*(y1 - y2)) <= d * d)
return 1;
else return 0;
}
void init(int x)
{
up(i, 0, x)
{
fr[i] = i;
ht[i] = 1;
}
}
int find(int x)
{
if (fr[x] == x)
return x;
else return fr[x] = find(fr[x]);
}
void unite(int x, int y)
{
x = find(x);
y = find(y);
if (x == y) return;
if (ht[x] < ht[y])
{
fr[x] = y;
}
else
{
fr[y] = x;
if (ht[x] == ht[y])
ht[x]++;
}
}
bool same(int x, int y)
{
if (find(x) == find(y))
return true;
else return false;
}
void putvy() {
cout << "SUCCESS" << endl;
}
void putfl() {
cout << "FAIL" << endl;
}
int main()
{
cin >> n >> d;
up(i, 0, n)
{
cin >> cmp[i].x >> cmp[i].y;
}
memset(ok, 0, sizeof(ok));
up(i, 0, n)//预处理
{
up(j, i + 1, n)
{
if (distance(cmp[i].x, cmp[i].y, cmp[j].x, cmp[j].y))
{
ok[i][j] = 1;
ok[j][i] = 1;
}
}
}
init(n);
char ch;
int temp;
int temp2;
while (cin >> ch)//每一次询问,注意这里我是从0-n-1的并查集,输入的是1-n的询问
{
if (ch == 'O')
{
cin >> temp;
use[temp - 1] = 1;
up(i, 0, n)
{
if (i != temp - 1&&use[i])
{
if (ok[i][temp - 1] || ok[temp - 1][i])
{
unite(i, temp - 1);
}
}
}
}
else
{
cin >> temp >> temp2;
if (same(temp-1, temp2-1))
{
putvy();
}
else
putfl();
}
}
return 0;
}