245. 你能回答这些问题吗——线段树

本文详细介绍了一种高效的数据结构——线段树,用于解决单点修改和区间查询问题,特别是最大连续子段和的问题。文章提供了完整的实现代码,包括构建、更新和查询操作,并解释了关键数据成员的作用。

给定长度为 N 的数列 A,以及 M 条指令,每条指令可能是以下两种之一:

1 x y,查询区间 [x,y] 中的最大连续子段和,在这里插入图片描述

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,
−1000≤A[i]≤1000
输入样例:
5 3
1 2 -3 4 5
1 2 3
2 2 -1
1 3 2
输出样例:
2
-1

分析

  1. 首先此题是单点修改,区间查询问题,只需要pushup函数即可,然后分析所查询的是最大连续字段和tmax,想求根节点的tmax,分三种情况:①tmax仅在分界线左边取得,那么根节点的u.tmax直接等于左儿子的tmax即可;②tmax仅在分界线右边取得,那么根节点的u.tmax直接等于右儿子的tmax即可;③根节点的tmax跨越分界线(横跨区间),那么u.tmax由左儿子的最大后缀和t[l].rmax + 右儿子的最大前缀和组成t[r].lmax;(最大后缀和:以区间右端点为终点,方向向左的最大后缀和;最大前缀和:以区间做左端点开始,方向向右的最大前缀和)每个结点(包括每一个儿子)都有lmax,rmax; 所以根节点的tmax就是上面三种情况的max值;

在这里插入图片描述

  1. 想求tmax,用到了lmax,rmax,然后求得了tmax,那么如何求lmax,rmax,那么看下图;对于根节点来说,lmax分为两种情况:①从区间左端点开始,没越过分界线结束,那么根节点的lmax = 左儿子的lmax;②从区间左端点开始,越过分界线,那么根节点的lmax=左儿子的区间和(t[l].sum)+ 右儿子的最大前缀和(t[r].lmax);所以根节点的lmax取值为上面两种情况的max值;
    同理rmax的求法一样,也是上面两种情况,可以得到下图的结论;

在这里插入图片描述
3. 上面求lmax,rmax用到的sum怎么求呢,直接由左儿子、右儿子的sum加起来就行;
在这里插入图片描述
4. 那么通过上面3条对根节点的分析,可以求得tmax,那对于根的左儿子,他也算一个根,同样的做法来求,递归进行即可;
5. 这题由于有用于建立线段树的初始数组a,所以建树时候要注意初始化叶子结点,并且要pushup一下父节点,凡是修改了叶子结点,包括赋值,都要顺带给父亲更新一下;然后推荐一个大佬的讲解,讲的非常好:AcWing 245. 你能回答这些问题吗(线段树)

分析

#include <bits/stdc++.h>

using namespace std;

const int N = 500010;
struct node {
    int l, r;
    int tmax; //最大连续子段和
    int lmax; //最大前缀和(从区间左端点出发)
    int rmax; //最大后缀和(从区间右端点出发)
    int sum; //区间和
};
int n, m;
int a[N];
node t[4 * N];

void pushup(node &u, node &l, node &r) {
    //根据上面推导的结论求u的 tmax,lmax,rmax,sum
    u.tmax = max(max(l.tmax, r.tmax), l.rmax + r.lmax);
    u.lmax = max(l.lmax, l.sum + r.lmax);
    u.rmax = max(r.rmax, r.sum + l.rmax);
    u.sum = l.sum + r.sum;
}

//儿子结点向上更新父亲结点
void pushup(int u) {
    pushup(t[u], t[u << 1], t[u << 1 | 1]);
}

void build(int u, int l, int r) {
    t[u].l = l, t[u].r = r;
    //到叶子结点了
    if (l == r) {
        //本题有用于建立线段树的初始数组a,所以这里需要赋个值,对于单一一个点,这几个值相等
        t[u].tmax = t[u].lmax = t[u].rmax = t[u].sum = a[l];
        return;
    }
    int mid = l + r >> 1;
    build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
    //本题有用于建立线段树的初始数组a,所以需要更新下父亲
    pushup(u);
}

void modify(int u, int x, int v) {
    //找到叶子结点
    if (t[u].l == t[u].r) {
        t[u].tmax = t[u].lmax = t[u].rmax = t[u].sum = v;
        return;
    }
    //找叶子结点
    int mid = t[u].l + t[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) {
    //1. [l,r]完全包含 u 的左右区间
    if (l <= t[u].l && r >= t[u].r)
        return t[u];
    int mid = t[u].l + t[u].r >> 1;
    //2. [l,r]完全在u的左儿子
    if (r <= mid)
        return query(u << 1, l, r);
        //3. [l,r]完全在u的右孩子
    else if (l >= mid + 1)
        return query(u << 1 | 1, l, r);
        //4. [l,r]横跨分界线(左右两边都有)
    else {
        auto left = query(u << 1, l, r);
        auto right = query(u << 1 | 1, l, r);
        node res;
        //计算res的tmax
        pushup(res, left, right);
        return res;
    }
}

int main() {
    cin >> n >> m;
    for (int i = 1; i <= n; ++i) {
        cin >> a[i];
    }
    build(1, 1, n);
    while (m--) {
        int k, x, y;
        cin >> k >> x >> y;
        if (k == 1) {
            if (x > y)
                swap(x, y);
            cout << query(1, x, y).tmax << endl;
        } else {
            modify(1, x, y);
        }
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

向上的yyy

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值