1208 Stars in Your Window
题目来源: Poj
基准时间限制:2 秒 空间限制:131072 KB 分值: 160 难度:6级算法题 收藏 关注
整点上有N颗星星,每颗星星有一个亮度。用一个平行于x轴和y轴,宽为W高为H的方框去套星星。套住的所有星星的亮度之和为S(包括边框上的星星),求S的最大值。
Input
第1行:共3个数N, W, H,中间用空格分割,N为星星的数量,W为方框的宽度,H为方框的高度。(2 <= N <= 50000, 1 <= W, H <= 10^9)
第2 - N + 1行:每行3个数,X, Y, L,中间用空格分隔,分别表示星星的横坐标X,纵坐标Y,以及星星的亮度L。(1 <= X, Y <= 10^9,1 <= L <= 10000)
Output
输出方框能够套住的最大亮度和S。
Input示例
6 3 3
1 1 2
2 2 3
3 3 4
4 4 3
5 5 2
6 6 1
Output示例
12
#include <stdio.h>
#include <bits/stdc++.h>
#define mod 1000000007
typedef long long ll;
using namespace std;
class P{
public:
ll x,x1,y,l;
};
int cmp(P a,P b){
if(a.y==b.y)a.l>b.l;
return a.y<b.y;
}
P s[100005];
ll a[100005],tree[8*50005],add[8*50005];
ll n,w,h,num,ans;
void push_down(int i){
tree[i*2]+=add[i];tree[i*2+1]+=add[i];
add[i*2]+=add[i];add[i*2+1]+=add[i];
add[i]=0;
}
void update(int i,int l,int r,int x,int y,int lg){
if(x<=l && y>=r){
add[i]+=lg;
tree[i]+=lg;
return;
}
int mid=l+r>>1;
push_down(i);
if(x<=mid) update(i*2,l,mid,x,y,lg);
if(y>mid) update(i*2+1,mid+1,r,x,y,lg);
tree[i]=max(tree[i*2],tree[i*2+1]);
}
int main()
{
cin>>n>>w>>h;
for(int i=1;i<=n;i++){
scanf("%lld%lld%lld",&s[i].x,&s[i].y,&s[i].l);
s[i].x1=s[i].x+w;s[i+n].x=s[i].x;
s[i+n].x1=s[i].x1;s[i+n].y=h+s[i].y;s[i+n].l=-s[i].l;
a[i]=s[i].x;a[i+n]=s[i].x1;
}
sort(a+1,a+1+n+n);
sort(s+1,s+1+n+n,cmp);
num=unique(a+1,a+1+n+n)-a-1;
for(int i=1;i<=2*n;i++){
int k1=lower_bound(a+1,a+1+num,s[i].x)-a;
int k2=lower_bound(a+1,a+1+num,s[i].x1)-a;
update(1,1,num,k1,k2,s[i].l);
ans=max(ans,tree[1]);
}
cout<<ans<<endl;
return 0;
}