洛谷3958 noip2017 奶酪

题目：奶酪

思路：并查集。

注意：一定要开long long，考场上就是这里挂了两个点。

代码：#include<bits/stdc++.h>
using namespace std;

#define ll long long

struct Hole{
ll x,y,z;
Hole(){}
Hole(ll xx,ll yy,ll zz){
x=xx,y=yy,z=zz;
}
};

ll n,h,r;
Hole a[1005];
ll fa[1005]={0};

void init(){
for(ll i=1;i<=n;i++) fa[i]=i;
}

ll find(ll x){
if(fa[x]==x) return x;
return fa[x]=find(fa[x]);
}

void Make(ll m,Hole s){
for(ll i=1;i<m;i++){
Hole t=a[i];

if(((double)t.x-s.x)*(t.x-s.x)+(t.y-s.y)*(t.y-s.y)+(t.z-s.z)*(t.z-s.z)<=4*(double)r*r){
ll f1=find(m),f2=find(i);
if(f1!=f2){
fa[f1]=f2;
}
}
}
}

bool checkUp(ll x){
if(a[x].z+r>=h&&a[x].z-r<=h) return true;
return false;
}

bool checkDown(ll x){
if(a[x].z+r>=0&&a[x].z-r<=0) return true;
return false;
}

bool Check(){
vector<ll> vec1,vec2;
for(ll i=1;i<=n;i++){
find(i);
if(checkUp(i)) vec1.push_back(i);
if(checkDown(i)) vec2.push_back(i);
}
for(ll i=0;i<vec1.size();i++){
for(ll j=0;j<vec2.size();j++){
ll x=vec1[i],y=vec2[j];
if(fa[x]==fa[y]) return true;
}
}
return false;
}

int main(){
ll T;
scanf("%lld",&T);
while(T--){
scanf("%lld%lld%lld",&n,&h,&r);
init();
for(ll i=1;i<=n;i++){
ll x,y,z;
scanf("%lld%lld%lld",&x,&y,&z);
a[i]=Hole(x,y,z);
Make(i,a[i]);
}
if(Check()) printf("Yes\n");
else printf("No\n");
}

return 0;
}