LA3938 线段树+分治

题意

 给出长度为n的整数序列,然后m次询问,对于每次询问,要求找到区间内的两个下标x,y使得区间和尽量大,如果有多解,x尽量小,还有多解,那么y也尽量小

分析

    最大连续和的分治分治算法是解决此题的关键,构造一棵线段树,并查集维护四个值:区间和sum,最大连续和max_sub、最大前缀和max_pre、最大后缀和max_last。建树与修改时注意
    max_sub[o]=max(max(max_sub[o*2], max_sub[o*2+1]), max_last[o*2]+max_pre[o*2+1]),
    同时应注意,查询时传递的时结点类型。
    //代码中顺便写了修改操作

代码

#include<cstdio>
#include<iostream>
#include<algorithm>
#define maxn 100010
using namespace std;

struct Tree {
    int max_sub, max_pre, max_last, sum; 
} T[3*maxn];

int array[maxn]; 

void build_tree(int o, int L, int R) {
    if(L == R) {
        if(array[L] >= 0) T[o].max_last = T[o].max_pre = T[o].max_sub = array[L];
        T[o].sum = array[L];
        return;
    }
    int M = L + (R-L)/2;
    build_tree(o*2, L, M);
    build_tree(o*2+1, M+1, R);
    T[o].sum = T[o*2+1].sum + T[o*2].sum;
    T[o].max_pre = max(T[o*2].max_pre, T[o*2].sum+T[o*2+1].max_pre);
    T[o].max_last = max(T[o*2+1].max_last, T[o*2+1].sum+T[o*2].max_last);
    T[o].max_sub = max(T[o*2].max_sub, T[o*2+1].max_sub);  
    T[o].max_sub = max(T[o].max_sub, T[o*2].max_last+T[o*2+1].max_pre);
}

Tree query(int o, int l, int r, int LL, int RR) {
    if(l >= LL && r <= RR) return T[o];
    if(l == r) return T[o];
    int M = l + (r-l)/2;
    Tree x = query(o*2, l, M, LL, RR), y = query(o*2+1, M+1, r, LL, RR);
    if(RR <= M) return x;
    if(LL >= M+1) return y;
    Tree ans;
    ans.sum = x.sum + y.sum;
    ans.max_pre = max(x.max_pre, x.sum+y.max_pre);
    ans.max_last = max(y.max_last, y.sum+x.max_last);
    ans.max_sub = max(x.max_sub, y.max_sub); 
    ans.max_sub = max(ans.max_sub, x.max_last+y.max_pre);
    return ans; 
}

void change(int o, int L, int R , int p, int v) {
    if(L == R) {
        if(array[L] > 0) T[o].max_last = T[o].max_pre = T[o].max_sub = v;
        T[o].sum = v;
        return;
    }
    int M = L + (R-L)/2;
    if(p <= M) change(o*2, L, M, p, v);
    if(p > M) change(o*2+1, M+1, R, p, v);
    T[o].sum = T[o*2+1].sum + T[o*2].sum;
    T[o].max_pre = max(T[o*2].max_pre, T[o*2].sum+T[o*2+1].max_pre);
    T[o].max_last = max(T[o*2+1].max_last, T[o*2+1].sum+T[o*2].max_last);
    T[o].max_sub = max(T[o*2].max_sub, T[o*2+1].max_sub); 
    T[o].max_sub = max(T[o].max_sub, T[o*2].max_last+T[o*2+1].max_pre);
}

int main() {
    int n;
    cin >> n;
    for(int i = 1; i <= n; i++) scanf("%d", &array[i]);
    build_tree(1, 1, n);
    int x, y;
    cin >> x >> y;
    cout << query(1, 1, n, x, y).max_sub;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值