NOIp2017D2T1(luogu3968) 奶酪 (并查集)

并查集。

判相切或相交的时候可以两边同时平方，就不需要double和开根号了。

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<vector>
 #include<queue>
 #include<cmath>
 #include<ctime>
 #define LL long long int
 #define inf 0x3f3f3f3f
 #define pw2(x) ((x)*(x))
 using namespace std;
 const int maxn=;

 LL rd(){
    LL x=;char c=getchar();int neg=;
    while(c<''||c>''){if(c=='-') neg=-;c=getchar();}
    while(c>=''&&c<='') x=x*+c-'',c=getchar();
    return x*neg;
 }

 LL pos[maxn][],r,h;
 int fa[maxn],N;

 int getf(int x){return x==fa[x]?x:fa[x]=getf(fa[x]);}
 void add(int a,int b){fa[getf(b)]=getf(a);}

 inline bool judge(LL x1,LL y1,LL z1,LL x2,LL y2,LL z2){
     return pw2(x1-x2)+pw2(y1-y2)+pw2(z1-z2)<=pw2(r)*;
 }

 int main(){
     int i,j,k;
     for(int t=rd();t;t--){
         N=rd(),h=rd(),r=rd();
         for(i=;i<=N+;i++) fa[i]=i;
         for(i=;i<=N;i++){
             pos[i][]=rd(),pos[i][]=rd(),pos[i][]=rd();
             if(fabs(pos[i][])<=r) add(,i);
             if(fabs(pos[i][]-h)<=r) add(N+,i);
         }for(i=;i<N;i++){
             for(j=i+;j<=N;j++){
                 if(judge(pos[i][],pos[i][],pos[i][],pos[j][],pos[j][],pos[j][])) add(i,j);
             }
         }
         if(getf()==getf(N+)) printf("Yes\n");
         else printf("No\n");
     }
     return ;
 }