2019杭电多校第六场 HDU6638 Snowy Smile（线段树区间合并+枚举）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6638

题意：给n个海盗箱，每个箱子都有一个二维坐标，现在画一个矩形，使矩形范围内的箱子价值和最大，求最大和。

 T(1≤T≤100)        n(1≤n≤2000)       xi,yi,wi(−1e9≤xi,yi,wi≤1e9)      ∑n≤10000.

思路：坐标离散化后，用线段树维护区间内最大值，总和，前缀最大值，后缀最大值，然后枚举上下界，取最大值即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define ls id<<1
#define rs id<<1|1
using namespace std;
typedef long long LL;
const int N=2e3+10;
int t,n;
vector<int>v[N];
int x[N],y[N],xx[N],yy[N],val[N];
struct node
{
    LL sum,maxx,maxl,maxr;
} tree[N<<2];
inline void build(int id,int l,int r)
{
    tree[id].maxx=tree[id].maxl=tree[id].maxr=tree[id].sum=0;
    if(l==r) return;
    int mid=(l+r)>>1;
    build(ls,l,mid);
    build(rs,mid+1,r);
}
inline void update(int id,int l,int r,int pos,int w)
{
    if(l==r)
    {
        tree[id].sum+=w;
        tree[id].maxl=tree[id].maxr=tree[id].maxx=tree[id].sum;
        return;
    }
    int mid=(l+r)>>1;
    if(pos<=mid)update(ls,l,mid,pos,w);
    if(pos>mid)update(rs,mid+1,r,pos,w);
    tree[id].maxl=max(tree[ls].maxl,tree[ls].sum+tree[rs].maxl);
    tree[id].maxr=max(tree[ls].maxr+tree[rs].sum,tree[rs].maxr);
    tree[id].maxx=max(max(tree[ls].maxx,tree[rs].maxx),tree[ls].maxr+tree[rs].maxl);
    tree[id].sum=tree[ls].sum+tree[rs].sum;
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(int i=1; i<=n; i++)
        {
            scanf("%d%d%d",&x[i],&y[i],&val[i]);
            xx[i]=x[i],yy[i]=y[i];
        }
        sort(xx+1,xx+1+n);
        sort(yy+1,yy+1+n);
        for(int i=1; i<=n; i++)            //离散化
        {
            x[i]=lower_bound(xx+1,xx+1+n,x[i])-xx;
            y[i]=lower_bound(yy+1,yy+1+n,y[i])-yy;
            v[y[i]].push_back(i);          //相同纵坐标放在一起
        }
        LL ans=0;
        for(int i=1; i<=n; i++)
        {
            build(1,1,n);                  //每枚举一个新的上界就重建线段树
            for(int j=i; j<=n; j++)
            {
                for(int k=0; k<v[j].size(); k++)
                {
                    update(1,1,n,x[v[j][k]],val[v[j][k]]);
                }
                ans=max(ans,tree[1].maxx);
            }
        }
        printf("%lld\n",ans);
         for(int i=1;i<=n;i++)           //对每个t都要清空
            v[i].clear();
    }
}

我爱学习