1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122
| #define MAX 10000 int todo[4 * MAX + 1]; int value[4 * MAX + 1];
void maintain(int root) { value[root] = value[root * 2] + value[root * 2 + 1]; }
void build(vector<int> &a , int root, int l, int r) { todo[root] = 0; if(l == r) { value[root] = a[l - 1]; return; } int m = (l + r) >> 1; build(a, root * 2, l, m); build(a, root * 2 + 1, m + 1, r); maintain(root); }
void do_(int root, int l, int r, int add) { value[root] += (r - l + 1) * add; todo[root] += add; }
void update(int root, int l, int r, int L, int R, int add) { if(L <= l && r <= R) { do_(root, l, r, add); return; } int m = (l + r) >> 1; if(todo[root]) { do_(root * 2, l, m, todo[root]); do_(root * 2 + 1, m + 1, r, todo[root]); todo[root] = 0; } if(m >= L) update(root * 2, l, m, L, R, add); if (m < R) update(root * 2 + 1, m + 1, r, L, R, add); maintain(root); }
int get(int root, int l, int r, int L, int R) { if (L <= l && r <= R) return value[root]; int rst = 0; int m = (l + r) >> 1; if (todo[root]) { do_(root * 2, l, m, todo[root]); do_(root * 2 + 1, m + 1, r, todo[root]); todo[root] = 0; } if (m >= L) rst += get(root * 2, l, m, L, R); if (m < R) rst += get(root * 2 + 1, m + 1, r, L, R); return rst; }
|