代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
int p[10005],rank[10005];

inline ll getint(){
    int f=1;
    ll ans=0;
    char c=getchar();
    while((c>'9'||c<'0')&&c!='-'){
        c=getchar();
    }
    if(c=='-'){
    	f=-1;
    	c=getchar();
    }
    do{
        ans=ans*10+c-'0';
        c=getchar();
    }while(c>='0'&&c<='9');
    //cout<<"getint:"<<ans<<endl;
    return ans*f;
}

inline int zx(int x){
    if(p[x]!=x)p[x]=zx(p[x]);
    return p[x];
}
inline void hb(int x,int y){
    x=zx(x);
    y=zx(y);
    if(rank[x]<rank[y])swap(x,y);
    p[y]=x;
    if(rank[x]==rank[y])++rank[x];
}
inline bool jh(int x,int y){
    return zx(x)==zx(y);
}

struct dong{
    ll x,y,z;
};
dong d[10000];
ll d1[10000],d2[10000];
int d1n=0,d2n=0;

ll dis(dong a,dong b){
    return (a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y)+(a.z-b.z)*(a.z-b.z);
}
signed main(){
    srand(time(NULL));
    int t=getint();
    while(t--){
    ////////////////////////////////////////////////////////
    memset(d1,0,sizeof(dong));
    memset(d2,0,sizeof(dong));
    memset(d,0,sizeof(dong));
    d1n=0;
    d2n=0;
    for(int i=0;i<10000;i++){
        p[i]=i;
    }
    ////////////////////////////////////////////////////////
    ll n=getint(),h=getint(),r=getint();
    for(int i=0;i<n;i++){
        d[i].x=getint();
        d[i].y=getint();
        d[i].z=getint();
        if(d[i].z+r>=h)d1[d1n++]=i;
        if(d[i].z-r<=0)d2[d2n++]=i;
        for(int j=0;j<i;j++){
            if(dis(d[i],d[j])<=4*r*r)hb(i,j);
        }
    }
    bool ok=0;
    for(int i=0;i<d1n;i++){
    	for(int j=0;j<d2n;j++){
    		if(jh(d1[i],d2[j]))ok=1;
    	}
    }
    if(ok)puts("Yes");
    else puts("No");
    ////////////////////////////////////////////////////////
    
    }
    return 0;
}
/*
1
2 5100500 1275250
0 0 1275250
0 50500 3825250
*/

评测结果