洛谷P1502 窗口的星星（线段树+扫描线+离散化）

洛谷P1502 窗口的星星（线段树+扫描线+离散化）

题意：
平面上有一些星星坐标为（x，y）权值为v，用一个长为H，宽为W的铁窗户框去框住尽量多的星星使得权值最大。铁边框上的不算在内。

思路

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
const int maxn=2e4+5;
ll sum[maxn<<4],lazy[maxn<<4],W,H,n,mp[maxn*4];
struct node{
	ll l,h,r,val;
	/*bool operator < (const node &x)const {
		//return (h!=x.h)?h<x.h:val>x.val;
		return h<x.h;
	}*/
}e[maxn*2];
bool cmp(node x,node y){
	if(x.h==y.h)return x.val>y.val;
	return x.h<y.h;
}
void pushdown(ll rt){
	if(lazy[rt]){
		sum[rt<<1]+=lazy[rt];
		sum[rt<<1|1]+=lazy[rt];
		lazy[rt<<1]+=lazy[rt];
		lazy[rt<<1|1]+=lazy[rt];
		lazy[rt]=0;
	}
}
void update(ll rt,ll l,ll r,ll x,ll y,ll val){
	if(x<=l&&r<=y){
		sum[rt]+=val;
		lazy[rt]+=val;
		return;
	}
	ll mid=(l+r)>>1;
	pushdown(rt);
	if(x<=mid)update(lson,x,y,val);
	if(y>mid)update(rson,x,y,val);
	sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
}
int main(){
	int T;
	scanf("%d",&T);
	while(T--){
		memset(sum,0,sizeof(sum));
		memset(lazy,0,sizeof(lazy));
		scanf("%lld%lld%lld",&n,&W,&H);
		for(int i=1;i<=n;i++){
			ll x,y,v;
			scanf("%lld%lld%lld",&x,&y,&v);
			e[i*2-1].l=e[i*2].l=x;
			e[i*2-1].r=e[i*2].r=x+W-1;
			e[i*2-1].h=y;
			e[i*2-1].val=v;
			mp[i*2-1]=x;
			mp[i*2]=x+W-1;
			e[i*2].val=-v;
			e[i*2].h=y+H-1;
		}
		sort(e+1,e+n*2+1,cmp);
		sort(mp+1,mp+1+2*n);
		ll k=unique(mp+1,mp+1+2*n)-mp-1;
		ll ans=0;
		for(int i=1;i<=2*n;i++){
			ll l=lower_bound(mp+1,mp+k+1,e[i].l)-mp-1;
			ll r=lower_bound(mp+1,mp+k+1,e[i].r)-mp-1;
			update(1,0,k,l,r,e[i].val);
			ans=max(ans,sum[1]);
		}
		printf("%lld\n",ans);
	}
}