1. 程式人生 > 其它 >線段樹的核心是區間 與query

線段樹的核心是區間 與query

https://www.acwing.com/activity/content/code/content/167568/

需要查詢的資訊是 l,r區間裡面的最大連續和
最大連續和由左右子節點l,r獲得 =max(l最大欄位和,r最大子段和,跨越中間的最大欄位和(=左節點的右端最大欄位和+右節點的左端最大欄位和))
所以一個節點要儲存的
l,r,lmax,rmax,ans
右因為 lmax的獲得需要對應左右兒子節點的l.sum+r.left
所以還要維護一個sum

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 500010;

int n, m;
int w[N];
struct Node
{
    int l, r;
    int sum, lmax, rmax, tmax;
}tr[N * 4];

void pushup(Node &u, Node &l, Node &r)//直接傳入本次要處理的節點和他的左右子節點 通過這樣獲得本次的節點資訊
{
    u.sum = l.sum + r.sum;
    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)
{
    tr[u] = {l, r};
    if (l == r) 
    {
        tr[u] = {l, r, w[r], w[r], w[r], w[r]};
        return ;
    }
    int mid = l + r >> 1;
    build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
    pushup(u);
    

}

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};
         return ;
    }
    
        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];
    
    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;
}