题意:水平、竖直方向各打三枪,每打一枪,该坐标该方向上的所有气球都会爆,但同一个方向不同枪要相隔r个单位,问最多可以射爆多少个气球。
题解:枚举打哪三行,然后用线段树求出打这三行的时候,竖直方向上最多射爆多少个气球。相加就能更新答案。线段树维护[l,r]区间内选三列气球来打,最多打多少个。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn = 1e5+5;
int x[maxn];
struct node
{
int l,r,mx;
}tr[maxn*4+1];
void build(int k,int l,int r)
{
tr[k].l = l,tr[k].r = r;
if(l == r)
return;
int m = (l+r)/2;
build(k*2,l,m);
build(k*2+1,m+1,r);
}
void update(int k,int p,int w)
{
if(tr[k].l == tr[k].r)
{
tr[k].mx += w;
return;
}
int m = (tr[k].l+tr[k].r)/2;
if(p<=m)
update(k*2,p,w);
else
update(k*2+1,p,w);
tr[k].mx = max(tr[k*2].mx,tr[k*2+1].mx);
}
vector<int>vec[maxn];
int main()
{
int n,r;
scanf("%d%d",&n,&r);
int mxx = 0,mxy = 0;
for(int i=1; i<=n; i++)
{
int a,b;
scanf("%d%d",&a,&b);
x[b]++;
vec[a].push_back(b);
mxx = max(a,mxx);
mxy = max(b,mxy);
}
build(1,0,mxy);
for(int i=0; i<=mxy; i++)
{
update(1,i,x[i]);
if(i+r<=mxy)
update(1,i,x[i+r]);
if(i+r*2<=mxy)
update(1,i,x[i+2*r]);
}
int tmp;
int ans = 0;
for(int i=0; i<=mxx; i++)
{
tmp = 0;
for(int j=0; j<vec[i].size(); j++)
{
tmp++;
update(1,vec[i][j],-1);
if(vec[i][j]-r>=0)
update(1,vec[i][j]-r,-1);
if(vec[i][j]-2*r>=0)
update(1,vec[i][j]-2*r,-1);
}
if(i+r<=mxx)
{
for(int j=0; j<vec[i+r].size(); j++)
{
tmp++;
update(1,vec[i+r][j],-1);
if(vec[i+r][j]-r>=0)
update(1,vec[i+r][j]-r,-1);
if(vec[i+r][j]-2*r>=0)
update(1,vec[i+r][j]-2*r,-1);
}
}
if(i+2*r<=mxx)
{
for(int j=0; j<vec[i+2*r].size(); j++)
{
tmp++;
update(1,vec[i+2*r][j],-1);
if(vec[i+2*r][j]-r>=0)
update(1,vec[i+2*r][j]-r,-1);
if(vec[i+2*r][j]-2*r>=0)
update(1,vec[i+2*r][j]-2*r,-1);
}
}
ans = max(ans,tmp+tr[1].mx);
for(int j=0; j<vec[i].size(); j++)
{
update(1,vec[i][j],1);
if(vec[i][j]-r>=0)
update(1,vec[i][j]-r,1);
if(vec[i][j]-2*r>=0)
update(1,vec[i][j]-2*r,1);
}
if(i+r<=mxx)
{
for(int j=0; j<vec[i+r].size(); j++)
{
update(1,vec[i+r][j],1);
if(vec[i+r][j]-r>=0)
update(1,vec[i+r][j]-r,1);
if(vec[i+r][j]-2*r>=0)
update(1,vec[i+r][j]-2*r,1);
}
}
if(i+2*r<=mxx)
{
for(int j=0; j<vec[i+2*r].size(); j++)
{
update(1,vec[i+2*r][j],1);
if(vec[i+2*r][j]-r>=0)
update(1,vec[i+2*r][j]-r,1);
if(vec[i+2*r][j]-2*r>=0)
update(1,vec[i+2*r][j]-2*r,1);
}
}
}
printf("%d",ans);
}