線段樹的核心是區間 與query
阿新 • • 發佈:2022-05-13
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; }