NOIP 2017 提高组 Day2 T1 奶酪 cheese.cpp

题目较为简单，将每个空洞看为一个点，N^2建图，BFS判断连通性即可。

//注意一个细节，accessible函数中，原本写

dx^2 + dy^2 + dz^2 <= r\*r\*4

是不行的（tyvj和cogs的数据过不了），会爆longlong.

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
const char Yes[5] = "Yes", No[5] = "No";

long long T, s, t, n, h, r;
long long x[1010], y[1010], z[1010];
bool vis[1010];
vector <long long> G[1010];
queue <long long> Q;

void init()
{
s = 0, t = 1001;
memset(vis, 0, sizeof vis);
for(int i = 0; i <= 1001; i++) G[i].clear();
}

void readInData()
{
scanf("%lld %lld %lld", &n, &h, &r);
for(int i = 1; i <= n; i++)
scanf("%lld %lld %lld", x+i, y+i, z+i);
}

long long sqr(long long num)
{
return num*num;
}

bool accessible(int a, int b)
{
return sqr(x[a]-x[b])+sqr(y[a]-y[b]) <= r*r*4 - sqr(z[a]-z[b]);
}

void add(int a, int b)
{
G[a].push_back(b);
G[b].push_back(a);
}

void buildGraph()
{
for(int i = 1; i <= n; i++)
{
if(r >= z[i]) add(s, i);
if(z[i]+r >= h) add(t, i);
for(int j = i+1; j <= n; j++)
if(accessible(i, j)) add(i, j);
}
}

void bfs()
{
Q.push(s);
vis[s] = 1;
while(!Q.empty())
{
int p = Q.front();
Q.pop();
for(int i = 0; i < G[p].size(); i++)
if(!vis[G[p][i]]) {
vis[G[p][i]] = 1;
Q.push(G[p][i]);
}
}
}

void solve()
{
init();
readInData();
buildGraph();
bfs();
printf("%s\n", vis[t]?Yes:No);
}

int main()
{
scanf("%lld", &T);
while(T--) solve();
return 0;
}

当然还有并查集版本，maybe更好写