这题我们用hash表来存储每一个位置有没有雷,将x和y转化为了一个数然后进行hash
观察到 r<=10
, 所以对于每一个火箭的的范围直接进行遍历,找该位置有没有雷,如果有雷进行dfs操作找完所有的雷,具体注释看代码
刚开始本来想到用并查集处理,但是并查集一般用于无向图,而这题地雷之间的关系是有向边的关系,所以不好用并查集
题目传送门
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 5e4+5, M = 499979;
//M 取到 N 的十倍提高hash效率降低时间复杂度
struct node{
int x,y,r;
}cir[N];
int h[M], id[M];
bool st[M];
int change(int x,int y)
{
return x * (int)(1e9+1) + y; //这里把x和y转换成一个数来哈希,值得注意的一个点是1e9+1需要转换成int,因为科学计数法是double类型的,最开始在这里wa了好久
//至于为什么是1e9+1,因为这样可以做到不重复,也可以另外找其他方法,只要做到不重复就可以。
}
int find(int x) // 这里用开放寻址法处理哈希冲突,即出现冲突一直往后推只到找到目标值,这里M开到了N的10倍,可以有效提高hash效率
{
int k = (x % M + M) % M; // 处理负数取模
while(h[k] != -1 && h[k] != x)
{
k++;
if(k == M) k = 0;
}
return k; //这里k返回两种含义,1:hash表中无x值并且x值需要存在h[k]中
//2:hash表中已存在x值,返回x值下标
}
int sqr(int x)
{
return x * x;
}
void dfs(int x,int y,int r) //dfs遍历半径所有的点,如果出现雷就继续dfs
{
int k = find(change(x,y));
st[k] = 1; //st数组标记该位置有雷并且已经被dfs过
for(int i=x-r;i<=x+r;i++)
{
for(int j=y-r;j<=y+r;j++)
{
if(sqr(i-x) + sqr(j-y) <= sqr(r))
{
int k = find(change(i,j));
if(id[k] && !st[k]) dfs(i,j,cir[id[k]].r);
//如果该位置有雷并且该雷未被dfs过就从该雷开始dfs
}
}
}
}
signed main()
{
ios::sync_with_stdio(false);cin.tie(nullptr);
int n,m;
cin>>n>>m;
memset(h, -1, sizeof h); // 初始化所有位置为无雷状态
for(int i=1;i<=n;i++)
{
int x,y,r;
cin>>x>>y>>r;
cir[i] = {x,y,r};
int k = find(change(x,y));
if(h[k] == -1) h[k] = change(x,y);
if(!id[k] || cir[id[k]].r < r) //这里处理相同坐标有多颗雷的问题,我们只保留半径最大的那颗雷
id[k] = i;
}
while(m--)
{
int x,y,r;
cin>>x>>y>>r;
for(int i=x-r;i<=x+r;i++)
{
for(int j=y-r;j<=y+r;j++)
{
if(sqr(i-x) + sqr(j-y) <= sqr(r))
{
int k = find(change(i,j));
if(id[k] && !st[k]) dfs(i,j,cir[id[k]].r);
}
}
}
}
int ans = 0;
for(int i=1;i<=n;i++)
{
if(st[find(change(cir[i].x, cir[i].y))]) ans++;
}
cout<<ans<<endl;
return 0;
}