洛谷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);
}
}