2019牛客暑期多校训练营（第十场）F Popping Balloons —— 线段树+枚举

This way

题意：

空间中有n个气球，你可以横着社三法子弹，竖着射三发子弹，且横着子弹的关系是y,y+r,y+2*r,竖着是x,x+r,x+2*r。问你怎么射才能射爆最多的气球。

题解：

我们枚举x轴的三发子弹，那么如何快速的找到y轴三发子弹的最大值，可以考虑将三个点缩成一个点，也就是将三个点的状态加到最下面这个点上。查询最大值即可

#include<bits/stdc++.h>
using namespace std;
const int N=3e5+5;
const int inf=1e5+1;
int mx[N*4];
void update(int l,int r,int root,int p,int v)
{
    if(l==r)
    {
        mx[root]+=v;
        return ;
    }
    int mid=l+r>>1;
    if(mid>=p)
        update(l,mid,root<<1,p,v);
    else
        update(mid+1,r,root<<1|1,p,v);
    mx[root]=max(mx[root<<1],mx[root<<1|1]);
}
vector<int>vec[N];
int r;
void up(int p,int v)
{
    update(1,inf,1,p,v);
    if(p-r>0)
        update(1,inf,1,p-r,v);
    if(p-r-r>0)
        update(1,inf,1,p-r-r,v);
}
int main()
{
    int n,x,y;
    scanf("%d%d",&n,&r);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&x,&y);
        x++,y++;
        vec[x].push_back(y);

        up(y,1);
    }
    int ans=0;
    for(int i=1;i<=inf;i++)
    {
        int sum=vec[i].size()+vec[i+r].size()+vec[i+2*r].size();
        for(int j=0;j<vec[i].size();j++)
            up(vec[i][j],-1);
        for(int j=0;j<vec[i+r].size();j++)
            up(vec[i+r][j],-1);
        for(int j=0;j<vec[i+2*r].size();j++)
            up(vec[i+r+r][j],-1);
            
        ans=max(ans,sum+mx[1]);

        for(int j=0;j<vec[i].size();j++)
            up(vec[i][j],1);
        for(int j=0;j<vec[i+r].size();j++)
            up(vec[i+r][j],1);
        for(int j=0;j<vec[i+2*r].size();j++)
            up(vec[i+r+r][j],1);
    }
    printf("%d\n",ans);
    return 0;
}