2019 杭电多校第6场 HDU - 6638 Snowy Smile 线段树最大子段和

最新推荐文章于 2019-08-29 10:43:00 发布

mmk27

最新推荐文章于 2019-08-29 10:43:00 发布

阅读量393

点赞数

CC 4.0 BY-SA版权

分类专栏：线段树

本文链接：https://blog.youkuaiyun.com/mmk27_word/article/details/98775461

线段树专栏收录该内容

68 篇文章

订阅专栏

本文介绍了一个算法问题，目标是在二维平面上找到一个矩形区域，使得该区域内所有点的权值之和达到最大。通过坐标离散化、点排序和线段树维护，实现了高效求解，时间复杂度为O(n^2logn)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：https://vjudge.net/problem/HDU-6638

题意：n个点，画一个矩形，使得矩形内的所有点的权值最大，求最大权值

题解：首先将坐标离散化到 O(n) 的范围内，方便后续的处理。将所有点按照y坐标排序，枚举矩形的上边界，然后往后依次加入每行的点，这样就确定了矩形的上下边界。线段树维护下每个点的权值，则答案即为最大连续子段和，。时间复杂度 O(n 2 log n)。维护的时候还要记录左连续，右连续，区间和，区间连续最大值，方便在pushup的时候使用。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2010;
struct Point {
    int x, y, w;
    bool operator < (const Point &b) const {
        if(y != b.y) return y < b.y;
        else return x < b.x;
    }
}p[N];
int n, b[N], len1, len2;
struct node {
    int l, r;
    ll maxx, lm, rm, sum;
}tree[N << 2];
void build(int l, int r, int cur) {
    tree[cur].l = l;
    tree[cur].r = r;
    tree[cur].maxx = tree[cur].lm = tree[cur].rm = tree[cur].sum = 0;
    if(l == r) return;
    int mid = (r + l) >> 1;
    build(l, mid, cur << 1);
    build(mid + 1, r, cur << 1 | 1);
}
void pushup(int cur) {
    tree[cur].maxx = max(0LL, max(tree[cur << 1].maxx, tree[cur << 1 | 1].maxx));
    tree[cur].maxx = max(tree[cur].maxx, max(0LL, tree[cur << 1].rm) + max(0LL, tree[cur << 1 | 1].lm));
    
    tree[cur].lm = max(0LL, max(tree[cur << 1].lm, tree[cur << 1].sum + tree[cur << 1 | 1].lm));
    tree[cur].rm = max(0LL, max(tree[cur << 1 | 1].rm, tree[cur << 1 | 1].sum + tree[cur << 1].rm));
    
    tree[cur].sum = tree[cur << 1].sum + tree[cur << 1 | 1].sum;
}
void update(int pos, int cur, int val) {
    if(tree[cur].l == tree[cur].r) {
        tree[cur].maxx += val;
        tree[cur].lm += val;
        tree[cur].rm += val;
        tree[cur].sum += val;
        return;
    }
    if(pos <= tree[cur << 1].r) update(pos, cur << 1, val);
    else update(pos, cur << 1 | 1, val);
    pushup(cur);
}
int main() {
    int T, pos;
    ll ans;
    scanf("%d", &T);
    while(T--) {
        scanf("%d", &n);
        len1 = len2 = 0;
        for(int i = 1; i <= n; i++) scanf("%d %d %d", &p[i].x, &p[i].y, &p[i].w), b[++len1] = p[i].x;
        sort(b + 1, b + 1 + len1);
        len1 = unique(b + 1, b + 1 + len1) - (b + 1);
        for(int i = 1; i <= n; i++) p[i].x = lower_bound(b + 1, b + 1 + len1, p[i].x) - b;

        for(int i = 1; i <= n; i++)    b[++len2] = p[i].y;
        sort(b + 1, b + 1 + len2);
        len2 = unique(b + 1, b + 1 + len2) - (b + 1);
        for(int i = 1; i <= n; i++) p[i].y = lower_bound(b + 1, b + 1 + len2, p[i].y) - b;
        sort(p + 1, p + 1 + n);    
        ans = 0;
        for(int i = 1; i <= len2; i++) {
            for(int j = n; j >= 1; j--) {
                if(p[j].y <= i) {
                    pos = j;
                    break;
                }
            }
            build(1, len1, 1);
            for(int j = i; j >= 1; j--) {
                while(pos && p[pos].y >= j) {
                    update(p[pos].x, 1, p[pos].w);
                    pos--;
                }
                ans = max(ans, tree[1].maxx);
            } 
            
        }    
        printf("%lld\n", ans);
    }
    return 0;
}