JZOJ 4218 【五校联考2day1】补给站

前言

此题出现在2016.1.26日的比赛中。当时因为在恶搞第三题，此题又没有想到如何判重，于是此题打完暴力就搁置了。%%%CTY大神~~~

题目描述

WYF为了保证他自己能够吃到足够多的牛排，来补充自己的脑力，所以他建了两个补给站，坐标分别为(ax,ay),(bx,by)。他有n个休息地点，第i个休息地点的坐标是(xi,yi)。每个补给站都有一个补给半径，当一个休息地点在以一个补给站为圆心，该补给站的补给半径为半径的圆中时（包括在圆周上），那个休息地点就会获得补给。现在有m个询问，每个询问会给出第一个补给站的补给半径r1和第二个补给站的补给半径r2,WYF想知道有多少个休息地点会得到补给。

数据范围

对于30%的数据:n≤5000,m≤5000.
对于100%的数据:n≤2*10^5,m≤10^5,ax,ay,bx,by,x,y∈[-100000,100000],r1,r2∈[0,300000]。

分析

题意可以简化为平面上有n个点，以另外两点为圆心，有m个询问，给出两圆的半径，求n个点中有多少被两个圆覆盖。
这道题若只考虑一个补给站，就变成大水题了。可以直接算出所有点到补给站的距离，排个序，对于每个询问，直接二分出被补给站半径覆盖到的点有多少个，时间复杂度$O(nlogn+mlogn)$。
现在回到两个补给站点的情况。我们设补给站点为A,B.如果仍用上述做法就会有一部分点既被r1覆盖又被r2覆盖。这样在算A,B覆盖的点时会算重。但是又没有很好的办法能在线处理重复，所以我们往离线方向去思考。
若把询问的r1排个序，这样r1是单调递增的，所以能被A覆盖的点数也是单调递增的。所以当r1增加时，可以算出新覆盖了多少个点。这样其实只用$O(m+n)$的复杂度。而对于r2则一样可以$O(logn)$二分出被覆盖的点数。
但是如何减去重复的点？
我们知道，重复的点其实就是同时能被两个圆覆盖的点。我们设数组$t[i]$表示与B的距离为i的点数。所以我们把新覆盖的点与B的距离加进数组，再减去$t[r2]$，就能很好的解决问题了。
$t$数组的实现用树状数组。这样就能以$O(logn)$的时间复杂度完成减重。总时间复杂度$O(nlogn+mlogn)$。
最后，强调一下：

离线大法好！
离线大法好！
离线大法好！

代码：

#include<cstdio>
#include<cmath>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int N=200010;
struct Point
{
    double x,y;
}a,b,c[N],d[N];
struct Q
{
    int r1,r2,pos;
}q[N/2];
double f[N];
int n,m,mx,tree[N*2],ans[N/2];
double sqr(double x){return x*x;}
double dis(Point a,Point b){return sqrt(sqr(a.x-b.x)+sqr(a.y-b.y));}
bool cmp1(Point a,Point b){return a.x<b.x;}
bool cmp2(Q a,Q b){return a.r1<b.r1 || a.r1==b.r1 && a.r2<b.r2;}
int lowbit(int x){return x&-x;}
int sum(int x)
{
    int t=0;
    for(;x;x-=lowbit(x)) t+=tree[x];
    return t;
}
void change(int x)
{
    for(;x<=mx;x+=lowbit(x)) tree[x]++;
}
int main()
{
    scanf("%d %d %lf %lf %lf %lf",&n,&m,&a.x,&a.y,&b.x,&b.y);
    fo(i,1,n)
    {
        scanf("%lf %lf",&c[i].x,&c[i].y);
        d[i].x=dis(a,c[i]);
        d[i].y=f[i]=dis(b,c[i]);
    }
    sort(d+1,d+n+1,cmp1);
    sort(f+1,f+n+1);
    fo(i,1,m)
    {
        scanf("%d %d",&q[i].r1,&q[i].r2);
        q[i].pos=i;
        mx=max(mx,max(q[i].r1,q[i].r2));
    }
    sort(q+1,q+m+1,cmp2);
    int last=0,j=0;
    fo(i,1,m)
    {
        while(d[j+1].x<=q[i].r1 && j<n) j++;
        int pos=upper_bound(f+1,f+n+1,q[i].r2)-f-1;
        fo(k,last+1,j)
        change(int(ceil(d[k].y)));//ceil:取上整
        ans[q[i].pos]=j+pos-sum(q[i].r2);
        last=j;
    }
    fo(i,1,m) printf("%d\n",ans[i]);
}