POJ2482-Stars in Your Window【线段树,扫描线,离散化】

正题

题目链接:http://poj.org/problem?id=2482

---

题目大意

有若干个点，每个点有不同的权值，求用一个h*w的矩阵扩起来的权值最大。

---

解题思路

先离散化一个坐标，然后另一个坐标就在$x\sim x+w-1$这个区域加上权值，然后每个点的高度一旦超过 h h <script type="math/tex" id="MathJax-Element-78">h</script>的范围就减去权值，然后问题就变成了找到一个权值最大的点了。
很像之前的宝石，不过要离散化

---

code

#include<cstdio>
#include<algorithm>
#define ll long long
#define N 10010
using namespace std;
struct treenode{
    ll l,r,w,lazy;
}a[8*N];
struct node{
    ll x,y,y2,w;
}d[2*N];
ll k,maxs,W,H,uqie[2*N],n,m;
void build(ll k,ll l,ll r)//建树
{
    a[k].w=a[k].lazy=0;
    a[k].l=l;a[k].r=r;
    if (l==r) return;
    ll mid=(l+r)/2;
    build(k*2,l,mid);
    build(k*2+1,mid+1,r);
}
void ddata(ll k)//下传标记
{
    a[k*2].w+=a[k].lazy;
    a[k*2+1].w+=a[k].lazy;
    a[k*2].lazy+=a[k].lazy;
    a[k*2+1].lazy+=a[k].lazy;
    a[k].lazy=0;
}
void updata(ll l,ll r,ll k,ll num)//区间修改
{
    if (uqie[a[k].l]==l&&uqie[a[k].r]==r)
    {
        a[k].w+=num;
        a[k].lazy+=num;
        return;
    }
    ddata(k);
    int mid=(a[k].l+a[k].r)/2;
    if (uqie[mid]>=r) updata(l,r,k*2,num);
    else if (uqie[mid]<l) updata(l,r,k*2+1,num);
    else updata(l,uqie[mid],k*2,num),updata(uqie[mid+1],r,k*2+1,num);
    a[k].w=max(a[k*2].w,a[k*2+1].w);
}
bool cmp(node x,node y)
{return x.x<y.x;}
int main()
{
    while(scanf("%lld%lld%lld",&n,&W,&H)==3)
    {
      ll xx,yy,ww,cnt=0;
      for (ll i=1;i<=n;i++)
      {
        scanf("%lld%lld%lld",&xx,&yy,&ww);
        d[++cnt]=(node){xx,yy,yy+H-1,ww};//加入的点
        uqie[cnt]=yy;
        d[++cnt]=(node){xx+W,yy,yy+H-1,-ww};//弹出的点
        uqie[cnt]=yy+H-1;
      }
      sort(uqie+1,uqie+1+cnt);
      sort(d+1,d+1+cnt,cmp);//坐标排序
      m=unique(uqie+1,uqie+1+cnt)-(uqie+1);//离散化
      build(1,1,m);maxs=0;
      for (ll i=1;i<=cnt;i++)
      {
        updata(d[i].y,d[i].y2,1,d[i].w);//区间修改
        maxs=max(maxs,a[1].w);//求最大值
      }
      printf("%lld\n",maxs);
    }
}