题目描述
小明最近迷上了一款名为《扫雷》的游戏。其中有一个关卡的任务如下， 在一个二维平面上放置着 n 个炸雷，第 i 个炸雷 (xi , yi ,ri) 表示在坐标 (xi , yi) 处存在一个炸雷，它的爆炸范围是以半径为 ri 的一个圆。
为了顺利通过这片土地，需要玩家进行排雷。玩家可以发射 m 个排雷火箭，小明已经规划好了每个排雷火箭的发射方向，第 j 个排雷火箭 (xj , yj ,rj) 表示这个排雷火箭将会在 (xj , yj) 处爆炸，它的爆炸范围是以半径为 rj 的一个圆，在其爆炸范围内的炸雷会被引爆。同时，当炸雷被引爆时，在其爆炸范围内的炸雷也会被引爆。现在小明想知道他这次共引爆了几颗炸雷？
你可以把炸雷和排雷火箭都视为平面上的一个点。一个点处可以存在多个炸雷和排雷火箭。当炸雷位于爆炸范围的边界上时也会被引爆。
输入格式
输入的第一行包含两个整数 n、m.
接下来的 n 行，每行三个整数 xi , yi ,ri，表示一个炸雷的信息。
再接下来的 m 行，每行三个整数 xj , yj ,rj，表示一个排雷火箭的信息。
输出一个整数表示答案。
样例输入
2 1
2 2 4
4 4 2
0 0 5
样例输出
2
提示
示例图如下，排雷火箭 1 覆盖了炸雷 1，所以炸雷 1 被排除；炸雷 1 又覆盖了炸雷 2，所以炸雷 2 也被排除。
对于 40% 的评测用例：0 ≤ x, y ≤ 109 , 0 ≤ n, m ≤ 103 , 1 ≤ r ≤ 10.
对于 100% 的评测用例：0 ≤ x, y ≤ 109 , 0 ≤ n, m ≤ 5 × 104 , 1 ≤ r ≤ 10.
第一种，图的深度优先遍历，邻接表实现， 由于点数有1e5,那么遍历所有图上的点是否联通，需要O(n^2)也就是需要2.5e9， 明显会超时。(WA)
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
#define int long long
using namespace std;
const int N=5e3+10;
bool st[N];
int n,m;
//存炸弹
struct node
{
int x,y,r;
}stu[N];
vector<int>v[N];
//判断是否在这颗雷是否在这个圆
bool sqr(int x,int y,int xx,int yy,int r)
{
if((xx-x)*(xx-x)+(yy-y)*(yy-y)<=r*r) return true;
return false;
}
//连成一个连通图 雷在这个雷的范围内的就扩展
void add(int idx)
{
int x=stu[idx].x,y=stu[idx].y,r=stu[idx].r;
for(int i=1;i<=n;i++)
{
if(i!=idx)
{
if(sqr(x,y,stu[i].x,stu[i].y,r)) v[idx].push_back(i);
}
}
}
//看有多少颗符合要求的炸弹
int dfs(int idx)
{
int sum=1;
st[idx]=1;
for(int i=0;i<v[idx].size();i++)
{
int j=v[idx][i];
if(!st[j])
{
st[j]=1;
sum+=dfs(j);
}
}
return sum;
}
//计算这颗排雷火箭能炸多少地雷
int dfs_Trave(int x,int y,int r)
{
int sum=0;
for(int i=1;i<=n;i++)
{
if(sqr(stu[i].x,stu[i].y,x,y,r))
{
if(!st[i])
sum+=dfs(i);
}
}
return sum;
}
signed main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
int x,y,r;cin>>x>>y>>r;
stu[i]={x,y,r};
}
for(int i=1;i<=n;i++)
{
add(i);
}
int sum=0;
for(int i=1;i<=m;i++)
{
int x,y,r;cin>>x>>y>>r;
sum+=dfs_Trave(x,y,r);
}
cout<<sum<<endl;
return 0;
}
AC版
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
#include<map>
#include<set>
#define int long long
using namespace std;
typedef pair<int,int>PII;
const int N=5e4+10;
bool st[N];
map<PII,int>mp;
int n,m,n1=0;
//存炸弹
struct node
{
int x,y,r,cnt;
bool operator < (node const & a) const
{
if(x!=a.x)
return x<a.x;
return y<a.y;
}
}stu[N];
/*
bool cmp(node xx,node yy)
{
if(xx.x==yy.x) return xx.y<yy.y;
return xx.x<yy.x;
}*/
vector<int>v[N];
//判断是否在这颗雷是否在这个圆
bool sqr(int x,int y,int xx,int yy,int r)
{
if((xx-x)*(xx-x)+(yy-y)*(yy-y)<=r*r) return true;
return false;
}
//连成一个连通图 雷在这个雷的范围内的就扩展
void add(int idx)
{
int x=stu[idx].x,y=stu[idx].y,r=stu[idx].r;
for(int i=idx-1;i>=0;i--)
{
if(r<(x-stu[i].x)) break;
if(sqr(x,y,stu[i].x,stu[i].y,r)) v[idx].push_back(i);
}
for(int i=idx+1;i<=n1;i++)
{
if(r<(stu[i].x-x)) break;
if(sqr(x,y,stu[i].x,stu[i].y,r)) v[idx].push_back(i);
}
}
//看有多少颗符合要求的炸弹
int dfs(int idx)
{
int sum=stu[idx].cnt;
st[idx]=1;
for(int i=0;i<v[idx].size();i++)
{
int j=v[idx][i];
if(!st[j])
{
st[j]=1;
sum+=dfs(j);
}
}
return sum;
}
//计算这颗排雷火箭能炸多少地雷
int dfs_Trave(int x,int y,int r)
{
node e1={x-r,y,r,1},e2={x+r,y,r,1};
int l1=lower_bound(stu+1,stu+1+n1,e1)-stu;
int r1=upper_bound(stu+1,stu+1+n1,e2)-stu;
int sum=0;
for(int i=l1;i<=r1;i++)
{
if(sqr(stu[i].x,stu[i].y,x,y,r))
{
if(!st[i])
sum+=dfs(i);
}
}
return sum;
}
signed main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
int x,y,r;cin>>x>>y>>r;
int id=mp[{x,y}];
if(id!=0)
{
stu[id].r=max(stu[id].r,r);
stu[id].cnt++;
}
else
{
int xx=1;
n1++;
stu[n1].x=x;
stu[n1].y=y;
stu[n1].r=r;
stu[n1].cnt=1;
mp[{x,y}]=n1;
}
}
sort(stu+1,stu+1+n1);
for(int i=1;i<=n1;i++)
{
add(i);
}
int sum=0;
for(int i=1;i<=m;i++)
{
int x,y,r;cin>>x>>y>>r;
sum+=dfs_Trave(x,y,r);
}
cout<<sum<<endl;
return 0;
}
