题目链接: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();
}
}
我爱学习