F. Popping Balloons （线段树做法） 2019暑假牛客多校 第十场

题目链接：

       https://ac.nowcoder.com/acm/contest/890/F

题目大意：

       n个气球在一个二维平面上，横着射三枪竖着射三枪，每枪距离是r，问最多能打多少个气球。

题目思路：

       可以说是暴力了。

       用一个vector数组存当前这一列的气球们的y坐标，再用一个a数组存 i 这行有多少个气球。

      y轴方向建立线段树，每行的单点值等于现在这行的气球数量+向下r+向下2*r 的数 量和。

       枚举x轴上的三枪最左边那枪的射击位置x，然后计算x ， x+r  ， x+2*r  ，三列一共有多少气球。

       枚举一个位置后，在线段树上把相应的气球删掉，意思是横轴打过了。计算完答案之后再加回来。

       一个气球最多被加减6次，所以总复杂度应该是6 * nlogn左右

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int MAXN = 1e5+5;
const int rMax = 100001;
vector<int>v[MAXN];
int a[4*MAXN];
int n,_r;
struct node
{
    int l,r,sum;
}C[4*MAXN];
void build(int p,int l,int r)
{
    C[p].l=l;C[p].r=r;
    if(l==r){
        C[p].sum=a[l]+a[l+_r]+a[l+_r+_r];
        return ;
    }
    int mid=(l+r)/2;
    build(2*p,l,mid);
    build(2*p+1,mid+1,r);
    C[p].sum=max(C[2*p].sum,C[2*p+1].sum);
}
void update(int p,int x,int k)
{
    if(C[p].l==C[p].r&&C[p].l==x){
        C[p].sum+=k;
        return ;
    }
    int mid=(C[p].l+C[p].r)/2;
    if(x<=mid)update(2*p,x,k);
    else update(2*p+1,x,k);
    C[p].sum=max(C[2*p].sum,C[2*p+1].sum);
}
void change(int y,int k)
{
    update(1,y,k);
    if(y-_r>0) update(1,y-_r,k);
    if(y-_r-_r>0) update(1,y-_r-_r,k);
}
int main()
{
    cin>>n>>_r;
    for(int i=1;i<=n;i++){
        int x,y;
        cin>>x>>y;
        x++,y++;
        v[x].push_back(y);
        a[y]++;
    }
    build(1,1,rMax);
    int Max=0;
    for(int i=1;i<=rMax;i++){
        int len=v[i].size();
        if(!len)continue;
        int now=v[i].size();
        for(int j=0;j<v[i].size();j++){
            change(v[i][j],-1);
        }
        if(i+_r<=rMax) {
            now+=v[i+_r].size();
            for(int j=0;j<v[i+_r].size();j++){
                change(v[i+_r][j],-1);
            }
        }
        if(i+_r+_r<=rMax) {
            now+=v[i+_r+_r].size();
            for(int j=0;j<v[i+_r+_r].size();j++){
                change(v[i+_r+_r][j],-1);
            }
        }
 
        now += C[1].sum;
 
        for(int j=0;j<v[i].size();j++){
            change(v[i][j],1);
        }
        if(i+_r<=rMax) {
            for(int j=0;j<v[i+_r].size();j++){
                change(v[i+_r][j],1);
            }
        }
        if(i+_r+_r<=rMax) {
            for(int j=0;j<v[i+_r+_r].size();j++){
                change(v[i+_r+_r][j],1);
            }
        }
        Max=max(Max,now);
    }
    cout<<Max<<endl;
}