NOIP2017提高组Day2第一题

Description

    现有一块大奶酪，它的高度为 h，它的长度和宽度我们可以认为是无限大的，奶酪中间有许多 半径相同 的球形空洞。我们可以在这块奶酪中建立空间坐标系，

在坐标系中，奶酪的下表面为z = 0，奶酪的上表面为z = h。
   现在，奶酪的下表面有一只小老鼠 Jerry，它知道奶酪中所有空洞的球心所在的坐标。两相切或是相交，则 Jerry 可以从其中一个空洞跑到另一个，

特别地，如果一个空洞与下表面相切或是相交，Jerry 则可以从奶酪下表面跑进空洞；

如果一个空洞与上表面相切或是相交，Jerry 则可以从空洞跑到奶酪上表面。
   位于奶酪下表面的 Jerry 想知道，在 不破坏奶酪 的情况下，能否利用已有的空洞跑到奶酪的上表面去?

 

Input

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

 

Output

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

 

Solutions

   深度优先搜索

代码

var
  boo:boolean;
  nm,n,h,r,d,k:longint;
  x,y,z:array [0..1001] of longint;
  bo:array [0..1001] of boolean;
procedure qsort(l,r:longint);
var
  i,j,t,mid:longint;
begin
  if l>r then exit;
  i:=l; j:=r;
  mid:=z[(l+r) div 2];
  repeat
    while z[i]<mid do inc(i);
    while z[j]>mid do dec(j);
    if i<=j then
      begin
        t:=x[i]; x[i]:=x[j]; x[j]:=t;
        t:=y[i]; y[i]:=y[j]; y[j]:=t;
        t:=z[i]; z[i]:=z[j]; z[j]:=t;
        inc(i); dec(j);
      end;
  until i>j;
  qsort(i,r);
  qsort(l,j);
end;

procedure init;
var
  i:longint;
begin
  fillchar(bo,sizeof(bo),true);
  boo:=true;
  readln(n,h,r);
  for i:=1 to n do
    readln(x[i],y[i],z[i]);
  d:=r+r;
  qsort(1,n);
end;

procedure search(m:longint);
var
  i:longint;
  t:real;
begin
  if not boo then exit;
  if z[m]+r>=h then
    begin
      boo:=false;
      exit;
    end;
  bo[m]:=false;
  for i:=1 to n do
    if bo[i] then
      begin
        t:=sqrt((x[i]-x[m])*(x[i]-x[m])+(y[i]-y[m])*(y[i]-y[m])+(z[i]-z[m])*(z[i]-z[m]));
        if t<=d then
          search(i);
      end;
end;

begin
  readln(nm);
  while nm>0 do
  begin
    init;
    for k:=1 to n do
      if (abs(z[k])<=r) and (bo[k]) then
        search(k);
    if boo then writeln('No')
           else writeln('Yes');
    dec(nm);
  end;
end.