Educational Codeforces Round 73 (Rated for Div. 2)F. Choose a Square(扫描线+线段树)

F. Choose a Square

题意：

输入$n(1\le n\le 5e5)$,表示有$n$个点，接下来$n$行，每行$x_i,y_i,c_i(0\le x_i,y_i\le 10^9,-10^6\le c_i\le 10^6)$。要求选取一个对角线再$y=x$上的一个正方形，使得正方形围成的$(\sum c_i)-l(正方形边长)$最大。

题解：

首先分析对于一个点$(x,y)$来说如何使他被正方形(左下角$(l,l)$,右上角$(r,r)$)包含，其实只要满足$l\le min(x,y)\le max(x,y)\le r$，那么不妨使所有的点$x\le y$,这样只要使$l\le x\le y\le r$。
然后把$x,y,c$按$x$从小到大排序，接下来扫一遍$x$,查询更新线段树就好。
线段树维护的是$pair$(最大值，最大值所对应的下标)；初始化线段树，一开始先把所有的$-r$加入线段树，对于$x$更新$y-maxn$加$c$,这样维护的就是$max((\sum c)-r)$,再把固定的$l$加上就是固定$l$时最大的答案。
时间复杂度$O(nlogn)$

代码：

#include<bits/stdc++.h>
#define ls(p) (p<<1)
#define rs(p) (p<<1|1)
#define ll long long
#define P pair<ll,ll>
using namespace std;
const int N=5e5+9;
int n;
struct T{int x,y,c;}a[N];
int x[N],nx;
ll ans,z,y;
bool cmp(T t1,T t2){return t1.x<t2.x;}
inline int getid(int y){return lower_bound(x+1,x+nx+1,y)-x;}
ll add[N<<2];
P mx[N<<2];
void push_down(int p){
    if(add[p]){
        add[ls(p)]+=add[p],add[rs(p)]+=add[p];
        mx[ls(p)].first+=add[p],mx[rs(p)].first+=add[p];
        add[p]=0;
    }
}
void push_up(int p){mx[p]=max(mx[ls(p)],mx[rs(p)]);}
void update(int p,int l,int r,int ql,int qr,int qz){
    if(ql<=l&&r<=qr){mx[p].first+=qz,add[p]+=qz;if(l==r)mx[p].second=l;}
    else{
        push_down(p);
        int mid=(l+r)>>1;
        if(ql<=mid)update(ls(p),l,mid,ql,qr,qz);
        if(qr>mid)update(rs(p),mid+1,r,ql,qr,qz);
        push_up(p);
     //   cout<<"yu"<<p<<" "<<mx[p].first<<" "<<mx[p].second<<endl;
    }
}
P query(int p,int l,int r,int ql,int qr){
    if(ql<=l&&r<=qr)return mx[p];
    else{
        push_down(p);
        int mid=(l+r)>>1;
        P ans=P(-2e9,0);
        if(ql<=mid)ans=query(ls(p),l,mid,ql,qr);
        if(qr>mid)ans=max(ans,query(rs(p),mid+1,r,ql,qr));
     //   cout<<"tu"<<p<<" "<<ans.first<<" "<<ans.second<<endl;
        return ans;
    }
}
/*void query(int p,int l,int r,int qx){
    if(l==r)cout<<"pu"<<l<<" "<<p<<" "<<mx[p].first<<" "<<mx[p].second<<endl;
    else{
        push_down(p);
        int mid=(l+r)>>1;
        query(ls(p),l,mid,qx);
        query(rs(p),mid+1,r,qx);
        cout<<"pu"<<p<<" "<<mx[p].first<<" "<<mx[p].second<<endl;
    }
}*/
int main(){
   // freopen("tt.in","r",stdin),freopen("tt.out","w",stdout);
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i].x>>a[i].y>>a[i].c;
        if(a[i].x>a[i].y)swap(a[i].x,a[i].y);
        x[++nx]=a[i].y;
    }
    sort(a+1,a+n+1,cmp);
   // for(int i=1;i<=n;i++)cout<<a[i].x<<" "<<a[i].y<<" "<<a[i].c<<endl;
    sort(x+1,x+nx+1),nx=unique(x+1,x+nx+1)-x-1;
    for(int i=1;i<=nx;i++)update(1,1,nx,i,i,-x[i]);
    for(int i=1;i<=n;i++)update(1,1,nx,getid(a[i].y),nx,a[i].c);
    //query(1,1,nx,1);
   // for(int i=1;i<=nx;i++)cout<<query(1,1,nx,i).first<<" "<<query(1,1,nx,i).second<<endl;
    z=y=2e9;
    for(int i=1;i<=n;i++){
       // query(1,1,nx,1);
        P zd=query(1,1,nx,getid(a[i].x),nx);
    //    cout<<"---"<<zd.first<<" "<<zd.second<<endl;
    //    cout<<a[i].x<<endl;
        if(ans<zd.first+a[i].x)ans=zd.first+a[i].x,z=a[i].x,y=x[zd.second];
        update(1,1,nx,getid(a[i].y),nx,-a[i].c);//cout<<a[i].y<<" "<<nx<<" "<<-a[i].c<<endl;
        while(i+1<=n&&a[i+1].x==a[i].x)i++,update(1,1,nx,getid(a[i].y),nx,-a[i].c);//cout<<a[i].y<<" "<<nx<<" "<<-a[i].c<<endl;
    }
    cout<<ans<<endl;
    cout<<z<<" "<<z<<" "<<y<<" "<<y<<endl;
    return 0;
}