题目
题意: 给定n个二维平面的地雷,给定m个导弹,可以引爆半径r以内的地雷,而半径r以内的地雷又可能引爆其半径r内的其他地雷。问最多能引爆多少个地雷。
思路: dfs或者bfs叭.
用unordered_map<ll,int> 记录对应位置的地雷,只需要记录对应坐标的地雷个数和最大半径即可。
因为unordered_map不支持用pair作key,可以做一个偏移,(1e9+1)x+y作为key,也可以唯一识别,至少比map快多了。
然后还可以减少点枚举,比如不用2r2r的枚举,x枚举2r,y从起点左右试探。
还有就是涨知识的是直接调用mp[x]会比较慢,不如判断mp.find(x) == mp.end()
时间复杂度: O(Πr*r(n+m)),还有哈希的常数。 代码源能过,acwing过不了,不管了,摆烂咧.
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N = 5e4+10;
#define mem(a,x) memset(a,x,sizeof(a))
#define fir(i,a,b) for(int i=a;i<=b;++i)
template<typename T>void write(T x)
{
if(x<0)
{
putchar('-');
x=-x;
}
if(x>9)
{
write(x/10);
}
putchar(x%10+'0');
}
template<typename T> void read(T &x)
{
x = 0;char ch = getchar();ll f = 1;
while(!isdigit(ch)){if(ch == '-')f*=-1;ch=getchar();}
while(isdigit(ch)){x = x*10+ch-48;ch=getchar();}x*=f;
}
int n,m,k,T;
ll fun(int x,int y)
{
return 1ll*(1e9+1)*x + 1ll*y;
}
struct node{
int x,y,r,tot;
}a[N];
ll ans = 0;
unordered_map<ll,int> mp;
bool vis[N];
inline void dfs(int i)
{
vis[i] = 1;
ans += a[i].tot;
int x = a[i].x,y = a[i].y,r = a[i].r;
for(int i=-r;i<=r;++i)
{
int tx = x + i;
ll dx = 1ll*(tx-x)*(tx-x);
for(int j=y;;++j)
{
ll dy = 1ll*(j-y)*(j-y);
if(dx+dy > 1ll*r*r) break;
ll wh = fun(tx,j);
if(mp.find(wh) != mp.end())
{
int t = mp[wh];
if(t && !vis[t])
{
dfs(t);
}
}
}
for(int j=y-1;;--j)
{
ll dy = 1ll*(j-y)*(j-y);
if(dx+dy > 1ll*r*r) break;
ll wh = fun(tx,j);
if(mp.find(wh) != mp.end())
{
int t = mp[wh];
if(t && !vis[t])
{
dfs(t);
}
}
}
}
}
int cnt = 0;
void solve()
{
read(n); read(m);
for(int i=0;i<n;++i)
{
int x,y,r; read(x),read(y),read(r);
ll wh = fun(x,y);
if(mp.find(wh) != mp.end())
{
int t = mp[wh];
a[t].r = max(a[t].r,r);
a[t].tot ++ ;
}
else
{
mp[wh] = ++cnt;
a[cnt] = {x,y,r,1};
}
}
while(m--)
{
int x,y,z; read(x),read(y),read(z);
a[cnt+1] = {x,y,z,0};
dfs(cnt+1);
}
write(ans);
// printf("%lld",ans);
}
int main(void)
{
solve();
return 0;
}