线段树:你能回答这些问题吗(单点修改 / 求最大连续区间和)

题目链接
题意:给定长度为N的数列A,以及M条指令,每条指令可能是以下两种之一:
1、“1 x y”,查询区间 [x,y] 中的最大连续子段和,即 maxx≤l≤r≤y{∑ri=lA[i]}。
2、“2 x y”,把 A[x] 改成 y。
对于每个查询指令,输出一个整数表示答案。
输入格式
第一行两个整数N,M。
第二行N个整数A[i]。
接下来M行每行3个整数k,x,y,k=1表示查询(此时如果x>y,请交换x,y),k=2表示修改。
输出格式
对于每个查询指令输出一个整数表示答案。
每个答案占一行。
数据范围
N≤500000,M≤100000
输入样例:
5 3
1 2 -3 4 5
1 2 3
2 2 -1
1 3 2
输出样例:
2
-1
思路:显然还是常规的查询和修改操作,只不过在建树的时候需要考虑创建哪些元素来实现目的:

  • 每个子区间内部的最大连续子段和tmax;
  • 如果父节点满足条件的最大连续子段和正好横跨两个子区间,那么就需要左儿子的最大后缀和rmax和右儿子的最大前嘴和lmax;
  • 如果父节点满足条件的最大区间不仅横跨,而且完全覆盖了左(右)儿子,那么我们还需要一个子区间的区间总和sum来实现。

当然在查询的时候也需要考虑是否有区间合并(横跨)的情况。
代码实现:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 5e5 + 5;
int n, m;
int w[N];
struct node
{
    int l, r;
    //区间的总和,最大前前缀和,最大后缀和,最大连续子区间和
    int sum, lmax, rmax, tmax;
}tr[N << 2];

void pushup(node &u, node &l, node &r)
{
    u.sum = l.sum + r.sum;
    //不横跨与横跨取个max
    u.lmax = max(l.lmax, l.sum + r.lmax);
    u.rmax = max(r.rmax, r.sum + l.rmax);
    u.tmax = max(max(l.tmax, r.tmax), l.rmax + r.lmax);
}

void pushup(int u)
{
    pushup(tr[u], tr[u << 1], tr[u << 1 | 1]);
}
//建树
void build(int u, int l, int r)
{
    //如果是叶节点,那么sum, lmax, rmax, tmax是相同的
    if(l == r) tr[u] = {l, r, w[r], w[r], w[r], w[r]};
    else{
        tr[u] = {l, r};
        int mid = l + r >> 1;
        build(u << 1, l, mid);
        build(u << 1 | 1, mid + 1, r);
        pushup(u);
    }
}
//单点修改,将w[x] -> y
void modify(int u, int x, int v)
{
    if(tr[u].l == x && tr[u].r == x)
        tr[u] = {x, x, v, v, v, v};
    else{
        int mid = tr[u].l + tr[u].r >> 1;
        if(x <= mid) modify(u << 1, x, v);
        else modify(u << 1 | 1, x, v);
        //更改节点值后需要更新它的所有父节点
        pushup(u);
    }
}
//查询的时候也可能会区间合并(查询的区间正好横跨两个子区间),所以直接返回节点
node query(int u, int l, int r)
{
    if(tr[u].l >= l && tr[u]. r <= r)
        return tr[u];
    else{
        int mid = tr[u].l + tr[u].r >> 1;
        //查询的区间正好是右(左)儿子区间,就遍历右(左)子区间
        if(r <= mid) return query(u << 1, l, r);
        else if(l > mid) return query(u << 1 | 1, l, r);
        else{
            //正好横跨两个子区间,就遍历两边
            auto left = query(u << 1, l, r);
            auto right = query(u << 1 | 1, l, r);
            node res;
            pushup(res, left, right);
            return res;
        }
    }
}

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i ++ )
        scanf("%d", &w[i]);
    build(1, 1, n);
    int k, x, y;
    while(m -- ){
        scanf("%d%d%d", &k, &x, &y);
        if(k == 1){
            //提醒交换
            if(x > y) swap(x, y);
            printf("%d\n", query(1, x, y).tmax);
        }
        else modify(1, x, y);
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值