HDU 6638 Snowy Smile（线段树求二维最大连续子段和）

http://acm.hdu.edu.cn/showproblem.php?pid=6638

先离散化坐标，枚举上下边界，线段树维护区间最大连续字段和。

#define bug(x) cout<<#x<<" is "<<x<<endl
#define IO std::ios::sync_with_stdio(0)
#include <bits/stdc++.h>
#define iter ::iterator
#define pa pair<int,ll>
using namespace  std;
#define ll long long
#define mk make_pair
#define pb push_back
#define se second
#define fi first
#define rs o<<1|1
#define ls o<<1
ll mod=998244353;
const int N=2e3+5;
int T,n;
int X[N],Y[N];
struct node{
    int x,y;
    ll w;
}a[N];
vector<pa>g[N];
ll sum[N*4],L[N*4],R[N*4],mx[N*4];
void push(int o){
    sum[o]=sum[ls]+sum[rs];
    L[o]=max(L[ls],sum[ls]+L[rs]);
    R[o]=max(R[rs],sum[rs]+R[ls]);
    ll t1=max(mx[ls],mx[rs]);
    ll t2=max(t1,R[ls]+L[rs]);
    mx[o]=0;
    mx[o]=max(mx[o],t2);
}
void up(int o,int l,int r,int k,ll w){
    if(l==r){
        sum[o]+=w;
        L[o]+=w;
        R[o]+=w;
        mx[o]+=w;
        return;
    }
    int m=(l+r)/2;
    if(k<=m)up(ls,l,m,k,w);
    else up(rs,m+1,r,k,w);
    push(o);
}
int main(){
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            scanf("%d%d%lld",&a[i].x,&a[i].y,&a[i].w);
            X[i]=a[i].x;
            Y[i]=a[i].y;
        }
        sort(X+1,X+1+n);
        sort(Y+1,Y+1+n);

        int sz1=unique(X+1,X+1+n)-X-1;
        int sz2=unique(Y+1,Y+1+n)-Y-1;

        for(int i=1;i<=n;i++){
            a[i].x=lower_bound(X+1,X+1+sz1,a[i].x)-X;
            a[i].y=lower_bound(Y+1,Y+1+sz2,a[i].y)-Y;
            g[a[i].y].pb(mk(a[i].x,a[i].w));
        }
        ll ans=0;
        for(int i=1;i<=sz2;i++){
            for(int j=1;j<=sz1*4;j++){
                sum[j]=L[j]=R[j]=mx[j]=0;
            }
            for(int j=i;j>=1;j--){
                for(int k=0;k<g[j].size();k++){
                    pa t=g[j][k];
                    up(1,1,sz1,t.fi,t.se);
                }
                ans=max(ans,mx[1]);
            }
        }
        printf("%lld
",ans);
        for(int i=1;i<=sz2;i++)g[i].clear();
    }
}