2019牛客暑期多校训练营（第十场）F Popping Balloons(线段树)

题目链接
大意：给你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;
}