POJ 2482 Stars in Your Window

线段树+离散化+扫描线

AC之后，又认真读了一遍题目，好文章。

#include<cstdio>
#include<map>
#include<algorithm>
using namespace std;

const int maxn=+;
int n;
long long W,H;
long long x[maxn],y[maxn],v[maxn];
struct seg
{
    long long X;
    long long Y1,Y2;
    long long val;
}s[*maxn];
struct SegTree
{
    long long MAX;
    long long add;
}segTree[*maxn];

map<long long,int>m;
long long M[maxn];
int k;

bool cmp(const seg&a,const seg&b)
{
    if(a.X==b.X) return a.val>b.val;
    return a.X<b.X;
}

void pushUp(int rt)
{
    segTree[rt].MAX=max(segTree[*rt].MAX,segTree[*rt+].MAX);
}

void pushDown(int rt)
{
    if(segTree[rt].add)
    {
        segTree[*rt].add+=segTree[rt].add;
        segTree[*rt+].add+=segTree[rt].add;
        segTree[*rt].MAX+=segTree[rt].add;
        segTree[*rt+].MAX+=segTree[rt].add;
        segTree[rt].add=;
    }
}

void build(int l,int r,int rt)
{
    segTree[rt].add=;
    segTree[rt].MAX=;
    if(l==r) return;

    int m=(l+r)/;
    build(l,m,*rt);
    build(m+,r,*rt+);
}

void update(long long val,int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        segTree[rt].add+=val;
        segTree[rt].MAX+=val;
        return;
    }

    pushDown(rt);
    int m=(l+r)/;
    if(L<=m) update(val,L,R,l,m,*rt);
    if(R>m)  update(val,L,R,m+,r,*rt+);
    pushUp(rt);
}

void lsh()
{
    m.clear();
    k=;
    for(int i=;i<=n;i++)
    {
        if(m[y[i]]==)
        {
            m[y[i]]=;
            M[k++]=y[i];
        }
        if(m[y[i]+H-]==)
        {
            m[y[i]+H-]=;
            M[k++]=y[i]+H-;
        }
    }
    sort(M,M+k);
    //m.clear();   这个地方比较奇葩。清空了会TLE。。。。
    for(int i=;i<k;i++) m[M[i]]=i+;
}

int main()
{
    //freopen("F:\\in.txt","r",stdin);
    while(~scanf("%d%lld%lld",&n,&W,&H))
    {
        for(int i=;i<=n;i++) scanf("%lld%lld%lld",&x[i],&y[i],&v[i]);
        lsh();
        int tot=;
        for(int i=;i<=n;i++)
        {
            s[tot].X=x[i];
            s[tot].Y1=m[y[i]];
            s[tot].Y2=m[y[i]+H-];
            s[tot].val=v[i];
            tot++;

            s[tot].X=x[i]+W-;
            s[tot].Y1=m[y[i]];
            s[tot].Y2=m[y[i]+H-];
            s[tot].val=-v[i];
            tot++;
        }

        sort(s,s+tot,cmp);

        build(,k,);

        long long max_val=;

        for(int i=;i<tot;i++)
        {
            update(s[i].val,s[i].Y1,s[i].Y2,,k,);
            max_val=max(segTree[].MAX,max_val);
        }
        printf("%lld\n",max_val);
    }
    return ;
}