离散化x,y坐标,就变成一个矩阵求最大子矩阵和的问题
子问题就是个动态维护区间最大子段和
线段树每个节点记录4个值,sum为区间和,lrmx为区间最大子段和,lmx从l开始向右的最大子段和,rmx从r开始向左的最大字段和
这4个值就可以每次pushup了。
然后枚举上边界和下边界,每次下边界+1就把这一行的不为0的点更新到线段树中,由于只有n个点,那么下边界一直移到最后,线段树也最多只会更新n次,总复杂度就是O(n^2logn)
tree[1].lrmx就是当前的答案。
#include<bits/stdc++.h>
#define maxl 2010
using namespace std;
int n,totx,toty;
int ax[maxl],ay[maxl],w[maxl];
long long ans;
long long a[maxl];
int numx[maxl],numy[maxl];
struct node
{
int l,r;
long long sum,lrmx;
long long rmx,lmx;
}tree[maxl*4];
vector <int> tmpy[maxl];
inline void prework()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d%d%d",&ax[i],&ay[i],&w[i]);
numx[i]=ax[i];numy[i]=ay[i];
}
sort(numx+1,numx+1+n);
totx=unique(numx+1,numx+1+n)-numx-1;
for(int i=1;i<=totx;i++)
tmpy[i].clear();
for(int i=1;i<=n;i++)
{
ax[i]=lower_bound(numx+1,numx+1+totx,ax[i])-numx;
tmpy[ax[i]].push_back(i);
}
sort(numy+1,numy+1+n);
toty=unique(numy+1,numy+1+n)-numy-1;
for(int i=1;i<=n;i++)
ay[i]=lower_bound(numy+1,numy+1+toty,ay[i])-numy;
}
inline void pushup(int k)
{
int ls=k<<1,rs=k<<1|1;
tree[k].sum=tree[ls].sum+tree[rs].sum;
tree[k].lrmx=max(tree[ls].lrmx,tree[rs].lrmx);
tree[k].lrmx=max(tree[k].lrmx,tree[ls].rmx+tree[rs].lmx);
tree[k].lmx=max(tree[ls].lmx,tree[rs].lmx+tree[ls].sum);
tree[k].rmx=max(tree[rs].rmx,tree[ls].rmx+tree[rs].sum);
}
inline void build(int k,int l,int r)
{
tree[k].l=l;tree[k].r=r;
if(l==r)
{
tree[k].sum=a[l];
tree[k].lrmx=tree[k].lmx=tree[k].rmx=max(0ll,a[l]);
return;
}
int mid=(l+r)>>1,ls=k<<1,rs=k<<1|1;
build(ls,l,mid);
build(rs,mid+1,r);
pushup(k);
}
inline void upd(int k,int l)
{
if(tree[k].l==tree[k].r)
{
tree[k].sum=a[l];
tree[k].lrmx=tree[k].lmx=tree[k].rmx=max(0ll,a[l]);
return;
}
int mid=(tree[k].l+tree[k].r)>>1;
if(l<=mid)
upd(k<<1,l);
else
upd(k<<1|1,l);
pushup(k);
}
inline void mainwork()
{
ans=0;int l,id;
for(int lowx=1;lowx<=totx;lowx++)
{
for(int i=1;i<=toty;i++)
a[i]=0;
build(1,1,toty);
for(int upx=lowx;upx<=totx;upx++)
{
l=tmpy[upx].size();
for(int i=0;i<l;i++)
{
id=tmpy[upx][i];
a[ay[id]]+=w[id];
upd(1,ay[id]);
}
ans=max(tree[1].lrmx,ans);
}
}
}
inline void print()
{
printf("%lld\n",ans);
}
int main()
{
int t;
scanf("%d",&t);
for(int i=1;i<=t;i++)
{
prework();
mainwork();
print();
}
return 0;
}