9.扫雷 - 蓝桥云课 (lanqiao.cn)

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1e5+100;
int n,m,res=0;
struct pt{
    int x,y,r;
};
typedef pair<int,int> pii;
map <pii,int> a;//炸雷的map,键是x,y,值是r
map <pii,int> b;//排雷火箭的map,键是x,y,值是r
map <pii,int> cnt;//炸雷数量的map,键是x,y,值是炸雷数量cnt
queue <pt> q;

void bfs()
{
    while(!q.empty())
    {
        auto t=q.front();
        q.pop();
        int tx=t.x,ty=t.y,tr=t.r;
        for(int i=tx-tr;i<=tx+tr;i++)
        {
            for(int j=ty-tr;j<=ty+tr;j++)
            {
                int dx=tx-i;
                int dy=ty-j;
                if(dx*dx+dy*dy>tr*tr||cnt[{i,j}]==0)//范围外的或者数量为0的
                    continue;
                res+=cnt[{i,j}];//该坐标雷的数量
                cnt.erase({i,j});//被引爆了的删掉该坐标
                q.push((pt){i,j,a[{i,j}]});//被引爆的雷入队
            }
        }
    }
}
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);cout.tie(nullptr);
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        int x,y,r;
        cin>>x>>y>>r;
        a[{x,y}]=max(a[{x,y}],r);//相同坐标记录半径大的
        cnt[{x,y}]++;
    }
    for(int i=1;i<=m;i++)
    {
        int x,y,r;
        cin>>x>>y>>r;
        b[{x,y}]=max(b[{x,y}],r);//相同坐标记录半径大的
    }
    for(auto i:b)//排雷火箭都入队
    {
        q.push((pt){i.first.first,i.first.second,i.second});
    }
    bfs();
    cout<<res;
    return 0;
}