HDU6638 Snowy Smile(二维离散化+线段树)

原题地址:http://acm.hdu.edu.cn/showproblem.php?pid=6638

题意:给出一个图,图上某些点藏有宝藏,可能是正数,可能是负数,现在需要你选择一个矩阵范围内的所有的宝藏,使得获得的权值和最大。

思路：由于宝藏的范围是$1e9$，所以很明显是需要离散化的,(比赛的时候突然不会写二维离散化了，囧)二维离散化只需要把x和y分别存起来，然后排序去重，对于原坐标中的每一个点，将它的$x，y$坐标$lowe{r}_{b}ound$即可。

离散化之后就相当于在一个最大$2000\ast 2000$的图中选择一个最大的子矩阵，这个问题其实就是经典的最大子矩阵问题，但是普通的解决方法的$n^3$的.考虑到这个题虽然是$200\ast 2000$个的点，但其实最多只有$2000$个点的地方有值，因此我们可以考虑每次枚举矩形的下界，每次添加一行矩阵的点后通过线段树来维护最大子段和，这样子复杂度就是$n^{2}logn$的。

PS：如果不会用线段树维护最大子段和，可以先去写一下这个题。

#include <bits/stdc++.h>

#define eps 1e-8
#define INF 0x3f3f3f3f
#define PI acos(-1)
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define CLR(x, y) memset((x),y,sizeof(x))
#define fuck(x) cerr << #x << "=" << x << endl;
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int seed = 131;
const int maxn = 1e5 + 5;
const int mod = 1e9 + 7;
int T, n;
struct node {
    int x, y;
    ll w;
} e[maxn];
int vx[maxn], vy[maxn];
ll mp[2004][2007];
ll sum[maxn << 2], mxleft[maxn << 2], mxright[maxn << 2], mxall[maxn << 2];
struct cur {
    ll sum, mxleft, mxright, mxall;
};

void push_up(int rt) {
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
    mxleft[rt] = max(mxleft[rt << 1], sum[rt << 1] + mxleft[rt << 1 | 1]);
    mxright[rt] = max(mxright[rt << 1 | 1], sum[rt << 1 | 1] + mxright[rt << 1]);
    mxall[rt] = max(max(mxall[rt << 1], mxall[rt << 1 | 1]), mxright[rt << 1] + mxleft[rt << 1 | 1]);
}

void build(int l, int r, int rt) {
    if (l == r) {
        sum[rt] = mxleft[rt] = mxright[rt] = mxall[rt] = 0;
        return;
    }
    int mid = (l + r) / 2;
    build(lson);
    build(rson);
    push_up(rt);
}


void update(int l, int r, int rt, int pos, ll val) {
    if (l == r) {
        sum[rt] += val;
        mxleft[rt] += val;
        mxright[rt] += val;
        mxall[rt] += val;
        return;
    }
    int mid = (l + r) / 2;
    if (pos <= mid) update(lson, pos, val);
    else update(rson, pos, val);
    push_up(rt);
}

cur query(int l, int r, int rt, int L, int R) {
    if (L <= l && R >= r) {
        return cur{sum[rt], mxleft[rt], mxright[rt], mxall[rt]};
    }
    int mid = (l + r) / 2;
    if (R <= mid) {
        return query(lson, L, R);
    } else if (L > mid) {
        return query(rson, L, R);
    } else {
        cur ans, t1, t2;
        t1 = query(lson, L, R);
        t2 = query(rson, L, R);
        ans.sum = t1.sum + t2.sum;
        ans.mxleft = max(t1.mxleft, t1.sum + t2.mxleft);
        ans.mxright = max(t2.mxright, t2.sum + t1.mxleft);
        ans.mxall = max(max(t1.mxall, t2.mxall), t1.mxright + t2.mxleft);
        return ans;
    }
}

typedef pair<int, int> pii;
vector<int> vec[2005];

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
#endif
    scanf("%d", &T);
    while (T--) {
        int N = 2005;
        CLR(mp, 0);
        scanf("%d", &n);
        int cntx = 0;
        int cnty = 0;
        for (int i = 1; i <= n; i++) {
            scanf("%d%d%lld", &e[i].x, &e[i].y, &e[i].w);
            vx[++cntx] = e[i].x;
            vy[++cnty] = e[i].y;
        }
        sort(vx + 1, vx + 1 + cntx);
        sort(vy + 1, vy + 1 + cnty);
        int lenx = unique(vx + 1, vx + 1 + cntx) - vx - 1;
        int leny = unique(vy + 1, vy + 1 + cnty) - vy - 1;
        for (int i = 1; i <= n; i++) {
            int px = lower_bound(vx + 1, vx + 1 + lenx, e[i].x) - vx;
            int py = lower_bound(vy + 1, vy + 1 + leny, e[i].y) - vy;
            mp[px][py] += e[i].w;
        }

        for (int i = 0; i <= lenx; i++) vec[i].clear();
        for (int i = 1; i <= lenx; i++) {
            for (int j = 1; j <= leny; j++) {
                if (mp[i][j] == 0) continue;
                vec[i].push_back(j);
            }
        }


        ll ans = 0;
        for (int i = 1; i <= lenx; i++) {
            build(1, N, 1);
            for (int j = i; j >= 1; j--) {
                int sz = vec[j].size();
                for (int k = 0; k < sz; k++) {
                    update(1, N, 1, vec[j][k], mp[j][vec[j][k]]);
                }
                ans = max(ans, mxall[1]);
            }
        }
        printf("%lld\n", ans);
    }
    return 0;
}