NOIp2017D2T1(luogu3968) 奶酪 (并查集)

并查集。

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

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 #include<vector>
 5 #include<queue>
 6 #include<cmath>
 7 #include<ctime>
 8 #define LL long long int
 9 #define inf 0x3f3f3f3f
10 #define pw2(x) ((x)*(x))
11 using namespace std;
12 const int maxn=1010;
13 
14 LL rd(){
15    LL x=0;char c=getchar();int neg=1;
16    while(c<'0'||c>'9'){if(c=='-') neg=-1;c=getchar();}
17    while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
18    return x*neg;
19 }
20 
21 LL pos[maxn][3],r,h;
22 int fa[maxn],N;
23 
24 int getf(int x){return x==fa[x]?x:fa[x]=getf(fa[x]);}
25 void add(int a,int b){fa[getf(b)]=getf(a);}
26 
27 
28 inline bool judge(LL x1,LL y1,LL z1,LL x2,LL y2,LL z2){
29     return pw2(x1-x2)+pw2(y1-y2)+pw2(z1-z2)<=pw2(r)*4;
30 }
31 
32 int main(){
33     int i,j,k;
34     for(int t=rd();t;t--){
35         N=rd(),h=rd(),r=rd();
36         for(i=0;i<=N+1;i++) fa[i]=i;
37         for(i=1;i<=N;i++){
38             pos[i][0]=rd(),pos[i][1]=rd(),pos[i][2]=rd();
39             if(fabs(pos[i][2])<=r) add(0,i);
40             if(fabs(pos[i][2]-h)<=r) add(N+1,i);
41         }for(i=1;i<N;i++){
42             for(j=i+1;j<=N;j++){
43                 if(judge(pos[i][0],pos[i][1],pos[i][2],pos[j][0],pos[j][1],pos[j][2])) add(i,j);
44             }
45         }
46         if(getf(0)==getf(N+1)) printf("Yes\n");
47         else printf("No\n");
48     }
49     return 0;
50 }