CH Round #46A 磁力块

还是一道好题的

对于一个磁石是否被吸引，有两个关键字：距离和质量。（二维偏序？？）

好像是很厉害的分块姿势，先按第一关键字排序，在块中按第二关键字排

进行bfs，对于当前磁石，有1~k-1个块是第一关键字全部小于等于当前磁石的，那么暴力从块首往后，找到第一个第二关键字大于当前磁石属性的，那么前面都捡走，以后可以从这里开始找。

暴力枚举第k个块找答案。

一个优化就是找k的时候直接比较当前块的最大值就行了，因为当前的属性肯定是>kmin，<kmax滴

（垃圾CH本机AC提交WA幸好最后我搞对了）

#include<cstdio>

#include<iostream>

#include<cstring>

#include<cstdlib>

#include<algorithm>

#include<cmath>

using namespace std;

typedef long long LL;

int n;

struct node{LL dis,m,p,r;}a[];bool v[];

bool cmp1(node n1,node n2){return n1.m<n2.m;}

bool cmp2(node n1,node n2){return n1.dis<n2.dis;}

int block,st[];

struct LIST

{

    LL p,r;

}list[];

int be[];LL mx[];

int findk(LL p)

{

    for(int i=;i<=block;i++)

        if(p<mx[i])return i;

    return block+;

}

int main()

{

    freopen("1.in","r",stdin);

    freopen("1.out","w",stdout);

    LL xx,yy,x,y;

    scanf("%lld%lld%lld%lld%d",&xx,&yy,&list[].p,&list[].r,&n);

    for(int i=;i<=n;i++)

    {

        scanf("%lld%lld%lld%lld%lld",&x,&y,&a[i].m,&a[i].p,&a[i].r);

        a[i].dis=(x-xx)*(x-xx)+(y-yy)*(y-yy);

    }

    sort(a+,a+n+,cmp1);

    block=(int(sqrt(double(n+))))+;

    for(int i=;i<=n;i++)st[i]=(i-)/block+;

    for(int i=;i<=block;i++)

    {

        int St=(i-)*block+,Ed=min(i*block,n);mx[i]=a[Ed].m;

        if(St<=Ed)sort(a+St,a+Ed+,cmp2);

    }

    for(int i=;i<=block;i++)be[i]=;

    memset(v,false,sizeof(v));

    int head=,tail=,ans=;

    while(head<tail)

    {

        LL p=list[head].p,r=list[head].r;

        int k=findk(p);

        for(int i=;i<=k-;i++)

        {

            for(int j=be[i];j<=block&&(i-)*block+j<=n;j++)

            {

                int u=(i-)*block+j;

                if(a[u].dis>r*r){be[i]=j;break;}

                {

                    if(v[u]==false)

                    {

                        v[u]=true;

                        ans++;

                        list[tail].p=a[u].p, list[tail].r=a[u].r;

                        tail++;

                    }

                }

            }

        }

        if(k!=block+)

        {

            for(int j=be[k];j<=block&&(k-)*block+j<=n;j++)

            {

                int u=(k-)*block+j;

                if(a[u].dis>r*r)break;

                else if(a[u].m<=p)

                {

                    if(v[u]==false)

                    {

                        v[u]=true;

                        ans++;

                        list[tail].p=a[u].p, list[tail].r=a[u].r;

                        tail++;

                    }

                }

            }

        }

        head++;

    }

    printf("%d\n",ans);

    return ;

}