刷题记录:牛客NC16417[NOIP2017]奶酪

传送门:牛客

题目描述:

现有一块大奶酪，它的高度为 h，它的长度和宽度我们可以认为是无限大的，奶酪中间有许多半径相同的球
形空洞。我们可以在这块奶酪中建立空间坐标系， 在坐标系中，奶酪的下表面为 z = 0，奶酪的上表面为 z = h。
现在， 奶酪的下表面有一只小老鼠 Jerry， 它知道奶酪中所有空洞的球心所在的坐标。如果两个空洞相切或
是相交，则 Jerry 可以从其中一个空洞跑到另一个空洞，特别地，如果一个空洞与下表面相切或是相交， 
Jerry 则可以从奶酪下表面跑进空洞； 如果一个空洞与上表面相切或是相交， Jerry 则可以从空洞跑到奶酪
上表面。
位于奶酪下表面的 Jerry 想知道， 在不破坏奶酪的情况下，能否利用已有的空洞跑到奶酪的上表面去?
输入:
3
2 4 1
0 0 1
0 0 3
2 5 1
0 0 1
0 0 4
2 5 2
0 0 2
2 0 4
输出:
Yes
No
Yes

当年NOIP的一道题,算是一道并查集的经典题吧,看懂题目之后思维难度也并不很高

主要思路:

1. 显然我们应该使用并查集来维护.并且因为每个空洞都是球体,并且假设两个球体相交的话等价于球心之间的距离小于半径之和.显然我们可以将每一个相交的球体都使用我们的并查集来合并.我们此时看一下n的范围,我擦,竟然才1000,N^2显然不成问题,因此我们直接枚举判断即可
2. 然而我们的jerry想要从底部到顶部,显然是需要一个空洞在底部(也就是Z坐标减去半径<=0),并且还需要一个空洞在顶部(也就是Z坐标加上半径>=h),并且两个空洞之间还需要一条合理的路径,但是在1中我们已经说了,我们可以使用并查集来维护每一个相交的球体,对于这个合法路径,显然就是直接判断底部的任何一个空洞和顶部的任何一个空洞的father一样即可(即在同一个并查集之中),这个显然也是容易判断的

注意:这道题需要开longlong

具体的代码部分:

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <stack>
#include <deque>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
#define ll_maxn 0x3f3f3f3f3f3f3f3f
const double eps=1e-8;
struct Node{
	ll x,y,z;
}node[maxn];
int fa[maxn];
int find(int a) {
	if(a==fa[a]) return a;
	else return fa[a]=find(fa[a]);
}
int Cnt1[maxn];
int Cnt2[maxn];
int main() {
	int T;
	T=read();ll n,h,r;
	while(T--) {
		int cnt1=0;int cnt2=0;
		memset(node,0,sizeof(node));
		memset(Cnt1,0,sizeof(Cnt1));
		memset(Cnt2,0,sizeof(Cnt2));
		n=read();h=read();r=read();
		for(int i=1;i<=n;i++) fa[i]=i;
		for(int i=1;i<=n;i++) {
			node[i].x=read();node[i].y=read();node[i].z=read();
			if(node[i].z-r<=0) {
				Cnt1[++cnt1]=i;
			}
			if(node[i].z+r>=h) {
				Cnt2[++cnt2]=i;
			}
		}
		for(int i=1;i<=n;i++) {
			for(int j=i+1;j<=n;j++) {
				ll X=node[i].x-node[j].x;
				ll Y=node[i].y-node[j].y;
				ll Z=node[i].z-node[j].z;
				if(X*X+Y*Y+Z*Z<=4*r*r) {
					if(find(i)!=find(j)) {
						fa[find(i)]=find(j);
					}
				}
			}
		}
		int flag=0;
		for(int i=1;i<=cnt1;i++) {
			for(int j=1;j<=cnt2;j++) {
				if(find(Cnt1[i])==find(Cnt2[j])) {
					flag=1;
					printf("Yes\n");
					break;
				}
			}
			if(flag==1) break;
		} 
		if(flag==0) printf("No\n");
	}
	return 0;
}