题目链接:传送门
描述
在一片广袤无垠的原野上,散落着N块磁石。每个磁石的性质可以用一个五元组(x,y,m,p,r)描述,其中x,y表示其坐标,m是磁石的质量,p是磁力,r是吸引半径。若磁石A与磁石B的距离不大于磁石A的吸引半径,并且磁石B的质量不大于磁石A的磁力,那么A可以吸引B。
小取酒带着一块自己的磁石L来到了这篇原野的(x0,y0)处,我们可以视为磁石L的坐标为(x0,y0)。小取酒手持磁石L并保持原地不动,所有可以被L吸引的磁石将会被吸引过来。在每个时刻,他可以选择更换任意一块自己已经获得的磁石(当然也可以是自己最初携带的L磁石)在(x0,y0)处吸引更多的磁石。小取酒想知道,他最多能获得多少块磁石呢?
输入格式
第一行五个整数x0,y0,pL,rL,N,表示小取酒所在的位置,磁石L磁力、吸引半径和原野上散落磁石的个数。
接下来N行每行五个整数x,y,m,p,r,描述一块磁石的性质。
输出格式
输出一个整数,表示最多可以获得的散落磁石个数(不包含最初携带的磁石L)。
样例输入
0 0 5 10 5
5 4 7 11 5
-7 1 4 7 8
0 2 13 5 6
2 -3 9 3 4
13 5 1 9 9
样例输出
3
数据范围与约定
对于30%的数据,1<=N<=1000。
对于100%的数据,1<=N<=250000,-10^9<=x,y<=10^9,1<=m,p,r<=10^9。
题解:
首先,假设手上有若干块磁铁,我要让我吸到的磁铁尽可能多,显然最简单的办法就是把手上的磁铁都拿来吸吸看,把能吸到的都吸过来,
所以可以用类似于BFS的形式,对手头的磁铁用一个队列维护,取出队头磁铁尝试吸引,把能吸到的磁铁都入队,反复如此知道队列为空,即可知道已经吸到了所有能被吸过来的磁铁。
由于考虑一个磁铁能不能被吸引过来,要看满足两个条件与否:
1、距离 ≤ 吸引半径
2、手头磁铁的磁力 ≥ 被吸引的磁铁的质量
对平面上所有磁铁按照与我的距离从近到远进行升序排序,分成 $T$ 块,显然每个分块内含有 $O(\frac{N}{T})$ 个磁铁,再对每个分块内的磁铁按照质量升序排序,
那么,对于当前的查询 $Q(p,r)$,对于吸引半径 $r$,必然存在一个整数 $k$,满足:
1、第 $1 \sim k$ 个分块中所有磁铁距离均小于等于 $r$;
2、第 $k+2$ 个分块中所有磁铁距离均大于 $r$,不可能被吸引;
分类讨论:
①对于满足距离条件的第 $1 \sim k$ 个分块,分别从每个块的块头开始遍历,直到某一个磁铁质量大于磁力 $p$ 时就停止,同时将该磁铁设为新的块头;
②对于不确定是否满足距离条件的第 $k+1$ 个分块,暴力枚举块内所有磁铁,判断是否能被吸引,能吸引的就取走即可。
时间复杂度:
对于①,对于某一个分块,由于其中每个磁铁都只会被取走一次,均摊复杂度为 $O(1)$,一次询问最多处理 $O(T)$ 个分块,因此每次询问时间复杂度 $O(T)$;
对于②,每次询问需要 $O(\frac{N}{T})$ 的暴力枚举。
①和②合起来就是 $O(T+\frac{N}{T})$,显然取 $T= \sqrt N$ 时最小,为 $O(\sqrt N)$;又最多有 $O(N)$ 次询问,总时间复杂度 $O(N \sqrt N)$。
AC代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=+;
const int maxt=+; struct Mg
{
ll x,y,m,p,r;
ll d;
}mg[maxn];
inline ll sqr(ll x){return x*x;}
inline ll dist(Mg a,Mg b){return sqr(a.x-b.x)+sqr(a.y-b.y);}
bool cmp1(Mg a,Mg b){return a.d<b.d;}
bool cmp2(Mg a,Mg b){return a.m<b.m;} int n;
int len,tot;
int L[maxt],R[maxt];
ll maxd[maxt];
int vis[maxn]; int main()
{
cin>>mg[].x>>mg[].y>>mg[].p>>mg[].r>>n;
mg[].r*=mg[].r;
for(int i=;i<=n;i++)
{
scanf("%lld%lld%lld%lld%lld",&mg[i].x,&mg[i].y,&mg[i].m,&mg[i].p,&mg[i].r);
mg[i].r*=mg[i].r;
mg[i].d=dist(mg[i],mg[]);
}
sort(mg+,mg+n+,cmp1); len=sqrt(n);
tot=;
for(int i=;i<=n;i+=len)
{
tot++;
L[tot]=i,R[tot]=min(i+len-,n);
maxd[tot]=mg[R[tot]].d;
sort(mg+L[tot],mg+R[tot]+,cmp2);
} int ans=;
memset(vis,,sizeof(vis));
queue<int> q;
q.push();
vis[]=;
while(!q.empty())
{
Mg &now=mg[q.front()]; q.pop(); ans++;
for(int i=;i<=tot;i++)
{
if(now.r>=maxd[i])
{
for(int &j=L[i];j<=R[i] && now.p>=mg[j].m;j++)
{
if(vis[j]) continue;
q.push(j);
vis[j]=;
}
}
else
{
for(int j=L[i];j<=R[i];j++)
{
if(vis[j]) continue;
if(now.r>=mg[j].d && now.p>=mg[j].m)
{
q.push(j);
vis[j]=;
}
}
break;
} }
} cout<<ans-<<endl;
}