这个题一看就是并查集的板子题。
记得在合并的时候开,不要开根号就行。
我当年考的时候是拿跑过去的。。。
附代码:
#include<iostream> #include<algorithm> #include<cstdio> #include<cstring> #define MAXN 1010 using namespace std; int n,S,T,fa[MAXN]; long long h,r; struct point{ long long x,y,z; }a[MAXN]; char tp[100000],*p1=tp,*p2=tp; char get_char(){ return p1==p2&&(p2=(p1=tp)+fread(tp,1,100000,stdin),p1==p2)?EOF:*p1++; } inline int read(){ int date=0,w=1;char c=0; while(c<'0'||c>'9'){if(c=='-')w=-1;c=get_char();} while(c>='0'&&c<='9'){date=date*10+c-'0';c=get_char();} return date*w; } inline long long dis(int i,int j){ return ((a[i].x-a[j].x)*(a[i].x-a[j].x)+(a[i].y-a[j].y)*(a[i].y-a[j].y)+(a[i].z-a[j].z)*(a[i].z-a[j].z)); } int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);} void uniun(int x,int y){x=find(x);y=find(y);if(x!=y)fa[x]=y;} void work(){ for(int i=1;i<=n;i++){ if(a[i].z-r<=0)uniun(S,i); if(a[i].z+r>=h)uniun(i,T); for(int j=i+1;j<=n;j++)if(dis(i,j)<=r*r*4)uniun(i,j); } if(find(S)==find(T))printf("Yes\n"); else printf("No\n"); } void init(){ memset(a,0,sizeof(a)); memset(fa,0,sizeof(fa)); n=read();h=read();r=read(); S=n+1;T=n+2; fa[S]=S;fa[T]=T; for(int i=1;i<=n;i++){ a[i].x=read();a[i].y=read();a[i].z=read(); fa[i]=i; } } int main(){ int t=read(); while(t--){ init(); work(); } return 0; }