【HDOJ6638】Snowy Smile（线段树）

题意：一个二维平面上有n个点，每个点的坐标是（x[i]，y[i]），权值是w[i]

求一个矩形使得其中所有点的权值和最大，输出权值和

n<=2e3，x[i]，y[i]，w[i]的绝对值<=1e9

思路：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef pair<ll,ll> Pll;
typedef vector<int> VI;
typedef vector<PII> VII;
#define N  110000
#define M  4100000
#define fi first
#define se second
#define MP make_pair
#define pi acos(-1)
#define mem(a,b) memset(a,b,sizeof(a))
#define rep(i,a,b) for(int i=(int)a;i<=(int)b;i++)
#define per(i,a,b) for(int i=(int)a;i>=(int)b;i--)
#define lowbit(x) x&(-x)
#define Rand (rand()*(1<<16)+rand())
#define id(x) ((x)<=B?(x):m-n/(x)+1)
#define ls p<<1
#define rs p<<1|1

const ll MOD=998244353,inv2=(MOD+1)/2;
      double eps=1e-6;
      int INF=1e9;

ll t1[N<<2],t2[N<<2],t3[N<<2],t4[N<<1];
int x[N],y[N],w[N],X[N],Y[N];
VII V[N];

int read()
{
   int v=0,f=1;
   char c=getchar();
   while(c<48||57<c) {if(c=='-') f=-1; c=getchar();}
   while(48<=c&&c<=57) v=(v<<3)+v+v+c-48,c=getchar();
   return v*f;
}

void clear(int l,int r,int p)
{
    t1[p]=t2[p]=t3[p]=t4[p]=0;
    if(l==r) return;
    int mid=(l+r)>>1;
    clear(l,mid,ls);
    clear(mid+1,r,rs);
}

void pushup(int p)
{
    t1[p]=t1[ls]+t1[rs];
    t2[p]=max(t2[ls],t1[ls]+t2[rs]);
    t3[p]=max(t3[rs],t1[rs]+t3[ls]);
    t4[p]=max(max(t4[ls],t4[rs]),t3[ls]+t2[rs]);
}

void update(int l,int r,int x,int v,int p)
{
    if(l==r)
    {
        t1[p]+=v;
        t2[p]=t3[p]=t4[p]=max(t1[p],0ll);
        return;
    }
    int mid=(l+r)>>1;
    if(x<=mid) update(l,mid,x,v,ls);
     else update(mid+1,r,x,v,rs);
    pushup(p);
}

int main()
{
    //freopen("1.in","r",stdin);
    int cas=read();
    while(cas--)
    {
        int n=read();
        X[0]=Y[0]=0;
        rep(i,1,n)
        {
            x[i]=read(),y[i]=read(),w[i]=read();
            X[++X[0]]=x[i];
            Y[++Y[0]]=y[i];
        }
        sort(X+1,X+X[0]+1);
        sort(Y+1,Y+Y[0]+1);
        X[0]=unique(X+1,X+X[0]+1)-X-1;
        Y[0]=unique(Y+1,Y+Y[0]+1)-Y-1;
        rep(i,1,n)
        {
            x[i]=lower_bound(X+1,X+X[0]+1,x[i])-X;
            y[i]=lower_bound(Y+1,Y+Y[0]+1,y[i])-Y;
        }
        rep(i,1,X[0]) VII().swap(V[i]);
        rep(i,1,n) V[x[i]].push_back(MP(y[i],w[i]));
        //printf("isok
");
        ll ans=0;
        rep(i,1,X[0])
        {
            clear(1,Y[0],1);
            rep(j,i,X[0])
            {
                for(int k=0;k<V[j].size();k++) update(1,Y[0],V[j][k].fi,V[j][k].se,1);
                ans=max(ans,t4[1]);
            }
        }
        printf("%I64d
",ans);
    }

    return 0;
}