POJ 4047 Garden(線段樹)
阿新 • • 發佈:2018-12-26
參考題解
有n個連續的一維上的點,每個點有一個美麗值。
每次有三種操作:
1、將x點的美麗值改為y。
2、將x、y交換
3、查詢[x y]間,連續k個點的美麗值總和的最大值。
這裡我是看的題解的做法,利用線段樹,葉子i表示[i i+k]美麗值總和,然後每個點維護一個區間最大值。每個位置向前取k個數,作為這個位置的結果,線段樹維護下每個區間的最大值即可,當改變一個位置p 的數時,把他所影響的區間(p,p+k-1)更新一下即可,查詢(x,y)區間時,查詢(x+k-1,y)的最大值即可。
#include <iostream> #include <algorithm> #include <cstring> #include <string> #include <cstdio> #define MAXN 222222 #define MAXM 22222 #define INF 1000000007 #define lch(x) x << 1 #define rch(x) x << 1 | 1 #define lson l, m, rt << 1 #define rson m + 1, r, rt << 1 | 1 using namespace std; int val[MAXN], t[MAXN], mx[4 * MAXN], cover[4 * MAXN]; int n, k, m; //上推狀態 void up(int rt) { mx[rt] = max(mx[lch(rt)], mx[rch(rt)]); } void down(int rt) { //如果是沒有記錄更改值,那麼直接跳過 if(!cover[rt]) return ; //否則當前節點的把更改標記推給左右孩子 cover[lch(rt)] += cover[rt]; cover[rch(rt)] += cover[rt]; //孩子的值進行更改 mx[lch(rt)] += cover[rt]; mx[rch(rt)] += cover[rt]; //去除自身標記 cover[rt] = 0; } //建樹 void build(int l, int r, int rt) { //先將自身的標記置為0 cover[rt] = 0; if(l == r) { mx[rt] = t[l]; return; } int m = (l + r) >> 1; build(lson); build(rson); up(rt); } void update(int L, int R, int l, int r, int rt, int v) //區間更新,v是增量 { if(L <= l && R >= r) { mx[rt] += v; cover[rt] += v; return; } down(rt); int m = (l + r) >> 1; if(m >= L) update(L, R, lson, v); if(m < R) update(L, R, rson, v); up(rt); } int query(int L, int R, int l, int r, int rt) { if(L <= l && R >= r) return mx[rt]; down(rt); int tmp = -INF; int m = (l + r) >> 1; if(m >= L) tmp = max(tmp, query(L, R, lson)); if(m < R) tmp = max(tmp, query(L, R, rson)); return tmp; } void change(int x, int y) { //因為是每個葉節點都是代表了長度為k的和,所以一個點的值進行改變之後,一連k個值都要改 int st = x - k + 1, ed = x; if(st < 1) st = 1; update(st, ed, 1, n, 1, y - val[x]); val[x] = y; } int main() { int T; cin >> T; while(T--) { scanf("%d%d%d", &n, &m, &k); for(int i = 1; i <= n; i++) scanf("%d", &val[i]); t[1] = 0; for(int i = 1; i <= k; i++) t[1] += val[i]; for(int i = 2; i <= n - k + 1; i++) t[i] = t[i - 1] - val[i - 1] + val[i - 1 + k]; n = n - k + 1; build(1, n, 1); int op, x, y; while(m--) { scanf("%d%d%d", &op, &x, &y); if(op == 0) change(x, y); else if(op == 1) { int tmp = val[x]; change(x, val[y]); change(y, tmp); } else if(op == 2) printf("%d\n", query(x, y - k + 1, 1, n, 1)); } } return 0; }