洛谷P3958 [NOIP2017 提高组] 奶酪题解

最新推荐文章于 2022-10-07 21:39:19 发布

q779

最新推荐文章于 2022-10-07 21:39:19 发布

阅读量585

点赞数

分类专栏： OI 文章标签：算法数据结构

本文链接：https://blog.csdn.net/qq_50332374/article/details/125743710

版权

OI 专栏收录该内容

240 篇文章 3 订阅

订阅专栏

洛谷P3958 [NOIP2017 提高组] 奶酪题解

题目链接：P3958 [NOIP2017 提高组] 奶酪

题意：

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

现在，奶酪的下表面有一只小老鼠 Jerry，它知道奶酪中所有空洞的球心所在的坐标。如果两个空洞相切或是相交，则 Jerry 可以从其中一个空洞跑到另一个空洞，特别地，如果一个空洞与下表面相切或是相交，Jerry 则可以从奶酪下表面跑进空洞；如果一个空洞与上表面相切或是相交，Jerry 则可以从空洞跑到奶酪上表面。

位于奶酪下表面的 Jerry 想知道，在不破坏奶酪的情况下，能否利用已有的空洞跑到奶酪的上表面去?

空间内两点 $P_1(x_1,y_1,z_1)$ 、 $P_2(x_2,y_2,z_2)$ 的距离公式如下：

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

对于 $100\%$ 的数据， $\le n \le 1\times 10^3$ ， $\le h , r \le 10^9$ ， $\le 20$ ，坐标的绝对值不超过 $10^9$ 。

基本思想使用并查集维护连通性

单独处理与边界相交的空洞，分别用数组 $c_1,c_2$ 记录

然后对于每个空洞两两判断是否相交，如果相交则并查集合并

时间复杂度 $O(Qn^2)$

代码：

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iomanip>
#include <random>
using namespace std;
#define int long long
#define INF 0x3f3f3f3f3f3f3f3f
#define N (int)(1e3+15)
#define pf(x) ((x)*(x))
int Q,n,h,r,cnt1,cnt2,x[N],y[N],z[N],c1[N],c2[N],f[N];
void init(int n){for(int i=1; i<=n; i++) f[i]=i;}
int find(int x){return f[x]==x?x:f[x]=find(f[x]);}
void merge(int u,int v){f[find(u)]=find(v);}
int _dis(int x1,int y1,int z1,int x2,int y2,int z2)
{return pf(x1-x2)+pf(y1-y2)+pf(z1-z2);}
int dis(int a,int b){return _dis(x[a],y[a],z[a],x[b],y[b],z[b]);}
signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    // freopen("check.in","r",stdin);
    // freopen("check.out","w",stdout);
    cin >> Q;
    while(Q--)
    {
        cin >> n >> h >> r;
        init(n); cnt1=cnt2=0;
        for(int i=1; i<=n; i++)
        {
            cin >> x[i] >> y[i] >> z[i];
            if(z[i]+r>=h) c1[++cnt1]=i;
            if(z[i]-r<=0) c2[++cnt2]=i;
            for(int j=1; j<i; j++)
                if(dis(i,j)<=4*r*r) merge(i,j);
        }
        int ok=0;
        for(int i=1; i<=cnt1&&!ok; i++)
            for(int j=1; j<=cnt2&&!ok; j++)
                if(find(c1[i])==find(c2[j]))ok=1;
        cout << (ok?"Yes":"No") << '\n';
    }
    return 0;
}