[Gym-102346A] 偷偷偷 并查集处理图（坐标）

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

const int maxn = 1e5+5;
const int mod=998244353;
int pre[maxn];

struct node{
    int x,y,l;
}s[maxn];

bool check(node a,node b){
    return ( (a.x-b.x)*(a.x-b.x) + (a.y-b.y)*(a.y-b.y) <= (a.l+b.l)*(a.l+b.l));
}

int find(int x){
    if(x!=pre[x]){
        pre[x] = find(pre[x]);
    }
    return pre[x];
}

void Union(int x,int y){
    int rx=find(x),ry=find(y);
    if(find(x) != find(y)){
        pre[rx] = ry;
    }
}

struct Node{
    int x,y,a,b;
}ans[maxn];

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n,m,k;
    cin>>m>>n>>k;
    for(int i=1; i<=k; i++){
        cin>>s[i].x>>s[i].y>>s[i].l;
    }
    for(int i=1; i<=k; i++){
        pre[i] = i;
    }
    for(int i=1; i<=k-1; i++){
        for(int j=i+1; j<=k; j++){
            if( check(s[i],s[j]) ){
                Union(i,j);
            }
        }
    }

    for(int i=1; i<=k; i++){
        if(s[i].x<=s[i].l){
            ans[ find(i) ].x = 1;
        }
        if(s[i].y<=s[i].l){
            ans[ find(i) ].y = 1;
        }
        if(m-s[i].x<=s[i].l){
            ans[ find(i) ].a = 1;
        }
        if(n-s[i].y<=s[i].l){
            ans[find(i)].b = 1;
        }
    }
    bool ok=0;
    for(int i=1; i<=k; i++){
        if(ans[i].x==1 && ans[i].y==1){
            ok=1;
        }
        if(ans[i].a==1 && ans[i].b==1){
            ok=1;
        }
        if(ans[i].x==1 && ans[i].a==1){
            ok=1;
        }
        if(ans[i].y==1 && ans[i].b==1){
            ok=1;
        }
        //cout<<ans[i].x<<" "<<ans[i].y<<endl;
    }
    if(ok) cout<<"N"<<endl;
    else cout<<"S"<<endl;
}