<更新提示>
<第一次更新>
<正文>
磁力块
Description
在一片广袤无垠的原野上,散落着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)处吸引更多的磁石。小取酒想知道,他最多能获得多少块磁石呢?
Input Format
第一行五个整数x0,y0,pL,rL,N,表示小取酒所在的位置,磁石L 磁力、吸引半径和原野 上散落磁石的个数。
接下来N 行每行五个整数x,y,m,p,r,描述一块磁石的性质。
Output Format
输出一个整数,表示最多可以获得的散落磁石个数(不包含最初携带的磁石L)。
Sample Input
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
Sample Output
3
解析
显然我们可以建立一个(bfs)框架,队列中存储着每一次已经拿到的磁石,然后每次尝试用队头的磁石去吸引其他磁石,能够吸引到其他磁石就把磁石加入到队尾,直到队列为空。
那么我们需要解决的问题就是快速地判断哪些磁石能够被吸引,而判断是否能够被吸引的有两个关键信息:"质量"和"距离",我们考虑如下的分块算法:
将(n)块石头按照质量排序,分为(sqrt n)段,每一段中再按距离排序。
那么对于每一次队首的磁石,一定存在一段满足:这段以前所有磁石质量都小于当前队首磁石的磁力,这段以后所有磁石质量都大于当前队首磁石的磁力。
对于前一部分,由于每一段内部已经按照距离重新排过序,那么可以被吸引的磁石就一定在这些段的左端,对每一段将吸引的磁石加入,并将该段的开头移动到没有被吸引的第一个位置即可。
对于这一段,直接暴力扫描,将能吸引的磁石处理掉即可。
对于前一种处理方式来说,均摊复杂度是(O(1))的,总复杂度为(O(sqrt n)),对于朴素扫描的的那一段来说,时间复杂度也是(O(sqrt n)),所以总的时间复杂度就是(O(nsqrt n))。
(Code:)
#include <bits/stdc++.h>
using namespace std;
const int N = 250020 , SIZE = 5200;
struct magnet
{
long long m,p,r,dis;
};
magnet a[N],st;
long long n,x_,y_,l[SIZE],r[SIZE],belo[N],T,size,ans,flag[N],Maxm[SIZE];
inline long long calc(long long x1,long long y1,long long x2,long long y2)
{
return (x1-x2) * (x1-x2) + (y1-y2) * (y1-y2);
}
inline void input(void)
{
scanf("%lld%lld%lld%lld%lld",&x_,&y_,&st.p,&st.r,&n);
st.m = 0 , st.dis = 0 , st.r *= st.r;
for (int i=1;i<=n;i++)
{
long long x,y,m,p,r;
scanf("%lld%lld%lld%lld%lld",&x,&y,&m,&p,&r);
a[i] = (magnet){m,p,r*r,calc(x,y,x_,y_)};
}
}
inline void setblocks(void)
{
T = sqrt(n) , size = n/T;
for (int i=1;i<=T;i++)
{
if ( i * size > n ) break;
l[i] = (i-1) * size + 1;
r[i] = i * size;
}
if ( r[T] < n ) T++ , l[T] = r[T-1] + 1 , r[T] = n;
for (int i=1;i<=T;i++)
for (int j=l[i];j<=r[i];j++)
belo[j] = i;
}
inline bool compare_m(magnet p1,magnet p2)
{
return p1.m < p2.m;
}
inline bool compare_d(magnet p1,magnet p2)
{
return p1.dis < p2.dis;
}
inline void init(void)
{
sort( a+1 , a+n+1 , compare_m );
for (int i=1;i<=T;i++)
Maxm[i] = a[r[i]].m , sort( a+l[i] , a+r[i]+1 , compare_d );
}
inline void bfs(void)
{
queue < magnet > q;
q.push(st);
while ( !q.empty() )
{
magnet t = q.front();
q.pop(); ans ++;
for (int i=1;i<=T;i++)
{
if ( Maxm[i] <= t.p )
{
for (int j=l[i];j<=r[i];j++)
{
if ( a[j].dis <= t.r )
{
if ( !flag[j] )
flag[j] = true , q.push(a[j]);
}
else {l[i] = j; break;}
if ( j == r[i] ) l[i] = r[i] + 1;
}
}
else
{
for (int j=l[i];j<=r[i];j++)
if ( !flag[j] && a[j].dis <= t.r && a[j].m <= t.p )
flag[j] = true , q.push(a[j]);
break;
}
}
}
}
int main(void)
{
input();
setblocks();
init();
bfs();
printf("%lld
",ans-1);
return 0;
}
<后记>