题目链接
大意:给你n个气球位置,现在你可以水平,竖直取至多三排,三列的气球,且每次相邻的行,距离不超过k,列也是如此
思路:我们按列(每个列号有三个列的贡献)建线段树,然后从小到大遍历最小的行,然后更新在这三行的列。然后查询直接查最大值即可。(注意要撤回更新操作)
细节见代码:
#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mp make_pair
#define pb push_back
using namespace std;
LL gcd(LL a,LL b){return b?gcd(b,a%b):a;}
LL lcm(LL a,LL b){return a/gcd(a,b)*b;}
LL powmod(LL a,LL b,LL MOD){LL ans=1;while(b){if(b%2)ans=ans*a%MOD;a=a*a%MOD;b/=2;}return ans;}
const int N = 5e5 +11;
int n,R;
vector<int>r[N],c[N];
LL T[N],C[N];
void build(int now,int l,int r){
if(l==r){
T[now]=C[l];
return ;
}
int mid=l+r>>1;
build(now<<1,l,mid);
build(now<<1|1,mid+1,r);
T[now]=max(T[now<<1],T[now<<1|1]);
}
void up(int now,int l,int r,int pos,int d){
if(pos<0)return ;
if(l==r){
T[now]+=d;
return ;
}
int mid=l+r>>1;
if(mid>=pos)up(now<<1,l,mid,pos,d);
else up(now<<1|1,mid+1,r,pos,d);
T[now]=max(T[now<<1],T[now<<1|1]);
}
int main(){
ios::sync_with_stdio(false);
cin>>n>>R;
for(int i=1;i<=n;i++){
int s,t;
cin>>s>>t;
r[s].pb(t);
c[t].pb(s);
}
for(int i=0;i<=100000;i++){
C[i]=c[i].size()+c[i+R].size()+c[i+2*R].size();
}
build(1,0,100000);
LL ans=0;
for(int i=0;i<=100000;i++){
LL res=r[i].size()+r[i+R].size()+r[i+2*R].size();
for(auto k:r[i]){
up(1,0,100000,k,-1);
up(1,0,100000,k-R,-1);
up(1,0,100000,k-2*R,-1);
}
for(auto k:r[i+R]){
up(1,0,100000,k,-1);
up(1,0,100000,k-R,-1);
up(1,0,100000,k-2*R,-1);
}
for(auto k:r[i+2*R]){
up(1,0,100000,k,-1);
up(1,0,100000,k-R,-1);
up(1,0,100000,k-2*R,-1);
}
ans=max(ans,res+T[1]);
for(auto k:r[i]){
up(1,0,100000,k,1);
up(1,0,100000,k-R,1);
up(1,0,100000,k-2*R,1);
}
for(auto k:r[i+R]){
up(1,0,100000,k,1);
up(1,0,100000,k-R,1);
up(1,0,100000,k-2*R,1);
}
for(auto k:r[i+2*R]){
up(1,0,100000,k,1);
up(1,0,100000,k-R,1);
up(1,0,100000,k-2*R,1);
}
}
cout<<ans<<endl;
return 0;
}