Educational Codeforces Round 73(Div 2) F. Choose a Square(线段树)

题目描述：

平面上有n个点，每个点有权值（可为负数），让你画一个正方形，使得这个正方形包含的点的权值和减去正方形的边长的值最大。这个正方形的左下角和右上角都要在直线y=x上。

解题思路：

注意到一个点(x,y)如果要被正方形(a, a, b, b)包含，则a <= min(x,y)且 b >= max(x,y).那么可以把问题转换成一维上的问题：
有n个区间[l,r]，每个区间有一个权值c，求一个区间[L,R]，使得[L,R]包含的小区间的权值和减去区间长度的值 最大。
我们先不管区间长度，先只让区间包含的权值和最大。那么可以从大到小枚举左端点，如果当前枚举的端点上有区间(l,r)，那么这个区间要更新右端点为[r,inf)的答案（区间加）。可以用线段树实现区间加法，维护区间最大值，维护最大值所在的下标。
现在我们要考虑减去区间长度这一部分。我们注意到，对于确定的区间[L,R],它的答案是$\sum c_i-R+L$,这个L与当前枚举的左端点有关，而这个R对于每个右端点来说都是固定的，所以我们可以建树的时候把这个-R的贡献加上，这样每次查询的就是$\sum c_i-R$的最大值了。

#include<bits/stdc++.h>
#define ll long long
#define lowbit(x) ((x)&(-(x)))
#define mid ((l+r)>>1)
#define lson rt<<1, l, mid
#define rson rt<<1|1, mid+1, r
#define P pair<int, ll>
using namespace std;
const int maxn = 1e6 + 50;
const ll inf = 0x3f3f3f3f3f3f3f3f;
ll mx[maxn<<2], lz[maxn<<2], mxid[maxn<<2];
int n;
int cc[maxn<<1], num;
int x[maxn], y[maxn];
ll a[maxn];
vector<P> q[maxn<<1];
void init(){
    num = 0;
    cin>>n;
    for(int i = 1; i <= n; ++i){
        scanf("%d%d%I64d", &x[i], &y[i], &a[i]);
        cc[++num] = x[i]; cc[++num] = y[i];
    }
    sort(cc+1,cc+1+num); num = unique(cc+1,cc+1+num) - cc - 1;
    for(int i = 1; i <= n; ++i){
        x[i] = lower_bound(cc+1,cc+1+num,x[i])-cc;
        y[i] = lower_bound(cc+1,cc+1+num,y[i])-cc;
        if(x[i] > y[i]) swap(x[i], y[i]);
        //cout<<"x:"<<x[i]<<" y:"<<y[i]<<endl;
        q[x[i]].push_back(P(y[i], a[i]));
    }
}
void up(int rt){
    if(mx[rt<<1] >= mx[rt<<1|1]) mx[rt] = mx[rt<<1], mxid[rt] = mxid[rt<<1];
    else mx[rt] = mx[rt<<1|1], mxid[rt] = mxid[rt<<1|1];
}
void down(int rt){
    lz[rt<<1] += lz[rt]; lz[rt<<1|1] += lz[rt];
    mx[rt<<1] += lz[rt]; mx[rt<<1|1] += lz[rt];
    lz[rt] = 0; return;
}
void build(int rt, int l, int r){
    if(l == r) {
        mx[rt] = -cc[l]; mxid[rt] = l;return;
    }build(lson); build(rson); up(rt);
    return;
}
void update(int rt, int l, int r, int L, int R, ll k){
    if(L <= l && r <= R){
        lz[rt] += k; mx[rt] += k;return;
    }down(rt);
    if(L <= mid) update(lson, L, R, k);
    if(R > mid) update(rson, L, R, k);
    up(rt);
    return;
}
ll cmx, cid;//查询用
void qry(int rt, int l, int r, int L, int R){
    if(L <= l && r <= R){
        if(mx[rt] > cmx) cmx = mx[rt], cid = mxid[rt];
        //cout<<"l:"<<l<<" r:"<<r<<" id:"<<mxid[rt]<<endl;
        return;
    }down(rt);
    if(L <= mid) qry(lson, L, R);
    if(R > mid) qry(rson, L, R);
}
void sol(){
    ll ans = 0;
    int cx = cc[num]+1, cy = cc[num]+1;
    build(1, 1, num);
    for(int i = num; i >= 1; --i){
        for(int j = 0; j < q[i].size(); ++j){
            int r = q[i][j].first;
            ll w = q[i][j].second;
            update(1, 1, num, r, num, w);
        }
        cmx = -inf; cid = -1;
        qry(1, 1, num, i, num);
        if(ans < cmx+cc[i]){
            ans = cmx+cc[i];
            cx = cc[i];
            cy = cc[cid];
        }
    }
    cout<<ans<<endl;
    cout<<cx<<" "<<cx<<" "<<cy<<" "<<cy<<endl;
}
int main()
{
	init();
	sol();
}