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=1010;

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

LL pos[maxn][3],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)*4;
}

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