1698 Just a Hook 線段樹區間更新
阿新 • • 發佈:2018-12-27
線段樹區間更新,有個懶惰標記,每次更新不進行到底,,用懶惰標記使得更新延遲到下次需要更新或者詢問到的時候。線段樹的精髓在於能不向下更新,儘可能利用子樹的根的資訊去獲取整棵子樹的資訊。
#include <iostream> #include <cstring> #include <cstdlib> #include <cstdio> #define lson l, m, rt << 1 #define rson m+1, r, rt << 1 | 1 using namespace std; const int MAXN = 200020; int col[MAXN<<2];//懶惰標記,每次更新到下一次需要更新的或者需要詢問的節點 int sum[MAXN<<2]; void PushUp(int rt) { sum[rt] = sum[rt<<1] + sum[rt<<1|1]; } void PushDown(int rt, int m)//鄉下更新一次,與懶惰標記一起理解 { if(col[rt])//col[rt]表明上一次已經更新到rt節點,現在需要向其子節點更新一次 { col[rt<<1] = col[rt<<1|1] = col[rt]; sum[rt<<1] = col[rt] * (m - (m>>1)); sum[rt<<1|1] = col[rt] * (m>>1); col[rt] = 0; } } void Build(int l, int r, int rt) { col[rt] = 0; sum[rt] = 1; if(l == r) return; int m = (l+r)>>1; Build(lson); Build(rson); PushUp(rt); } void UpData(int L, int R, int c, int l, int r, int rt) { if(L <= l && R >= r) { col[rt] = c; sum[rt] = c * (r-l+1); return; } PushDown(rt, r-l+1); int m = (l+r)>>1; if(L <= m) UpData(L, R, c, lson); if(R > m) UpData(L, R, c, rson); PushUp(rt); } int main() { int T, n, Q; int kcase = 1; scanf("%d", &T); while(T--) { scanf("%d%d", &n, &Q); Build(1, n, 1); while(Q--) { int a, b, c; scanf("%d%d%d",&a, &b, &c); UpData(a, b, c, 1, n, 1); } printf("Case %d: The total value of the hook is %d.\n",kcase++, sum[1]); } system("pause"); return 0; }