[NOIP2017]奶酪

现有一块大奶酪，它的高度为 h，它的长度和宽度我们可以认为是无限大的，奶酪中间有许多半径相同的球形空洞。我们可以在这块奶酪中建立空间坐标系，在坐标系中，奶酪的下表面为 z = 0，奶酪的上表面为 z = h。
现在，奶酪的下表面有一只小老鼠 Jerry，它知道奶酪中所有空洞的球心所在的坐标。如果两个空洞相切或是相交，则 Jerry 可以从其中一个空洞跑到另一个空洞，特别地，如果一个空洞与下表面相切或是相交， Jerry 则可以从奶酪下表面跑进空洞；如果一个空洞与上表面相切或是相交， Jerry 则可以从空洞跑到奶酪上表面。
位于奶酪下表面的 Jerry 想知道，在不破坏奶酪的情况下，能否利用已有的空洞跑到奶酪的上表面去?
空间内两点 P₁(x₁,y₁,z₁) 、P₂(x₂,y₂,z₂) 的距离公式如下：

$dist(P_1,P_2)=\sqrt{(x_1-x_2)^2+(y_1-y_2)^2+(z_1-z_2)^2}$

每个输入文件包含多组数据。
输入文件的第一行，包含一个正整数 T，代表该输入文件中所含的数据组数。
接下来是 T 组数据，每组数据的格式如下：
第一行包含三个正整数 n， h 和 r，两个数之间以一个空格分开，分别代表奶酪中空洞的数量，奶酪的高度和空洞的半径。
接下来的 n 行，每行包含三个整数 x, y, z，两个数之间以一个空格分开，表示空洞球心坐标为 (x,y,z)。

输出文件包含 T 行，分别对应 T 组数据的答案，如果在第 i 组数据中， Jerry 能从下表面跑到上表面，则输出“Yes”，如果不能，则输出“No”（均不包含引号）。

#include<bits/stdc++.h> using namespace std; int t,n,h,r,f[1234],x[1234],y[1234],z[1234],a; long long xx(int p){return 1ll*p*p;} int fu(int p){ if(f[p]==p)return p; return f[p]=fu(f[p]); } int main(){ cin>>t; while(t--){ cin>>n>>h>>r,a=0; for(int i=0;i<n;i++){ cin>>x[i]>>y[i]>>z[i];f[i]=i; for(int j=0;j<i;j++)if(xx(x[i]-x[j])+xx(y[i]-y[j])+xx(z[i]-z[j])<=4ll*r*r)f[fu(j)]=i; } for(int i=0;i<n;i++)for(int j=0;j<n;j++)if(fu(i)==fu(j)&&z[i]<=r&&z[j]>=h-r)a=1; if(a)puts("Yes"); else puts("No"); } }

详情