題目簡述: 有N個整數,Q次操作,每次操作為詢問一個區間[a, b]內數的和(0號操作)或者把一個區間內的數全部加上v(1號操作)
線段樹求解即可。
#include <cstdio> #include <algorithm> using std::min; using std::max; #define L(no) ((no) << 1) #define R(no) (L(no) | 1) #define PLUSTAG(no, val) plustag[no] += val, query[no] += val*(rb[no]-lb[no]+1) const int MAXN = 400001; int lb[MAXN], rb[MAXN], query[MAXN], ans, plustag[MAXN]; int sum(int a, int b) {return a + b;} void build(int no, int l, int r) {int mid = l + r >> 1;lb[no] = l;rb[no] = r;if(l == r) scanf("%d", &query[no]);else {build(L(no), l, mid);build(R(no), mid + 1, r);query[no] = query[L(no)] + query[R(no)];} } void getval(int no, int l, int r, int(*func)(int, int), int* key) {int mid = lb[no] + rb[no] >> 1;if(l <= r && (lb[no] <= r&& rb[no] >= r || rb[no] >= l && lb[no] <= l)) {if(lb[no] == l && rb[no] == r) ans = func(ans, key[no]);else {if(plustag[no]) {PLUSTAG(L(no), plustag[no]);PLUSTAG(R(no), plustag[no]);plustag[no] = 0;}getval(L(no), l, min(mid, r), func, key);getval(R(no), max(l, mid + 1), r, func, key);}} } void plus(int no, int l, int r, int val) {int mid = lb[no] + rb[no] >> 1;if(l <= r && (lb[no] <= r&& rb[no] >= r || rb[no] >= l && lb[no] <= l)) {if(lb[no] == l && rb[no] == r) {PLUSTAG(no, val);}else {if(plustag[no]) {PLUSTAG(L(no), plustag[no]);PLUSTAG(R(no), plustag[no]);plustag[no] = 0;}query[no] += (r - l + 1) * val;plus(L(no), max(l, lb[no]), min(mid, r), val);plus(R(no), max(l, mid + 1), min(r, rb[no]), val);}} } int main() {int n, q, i, op, a, b, v;scanf("%d%d", &n, &q);build(1, 1, n);for(i = 1; i <= q; i++) {scanf("%d%d%d", &op, &a, &b);if(op == 0) {ans = 0;getval(1, a, b, sum, query);printf("%d\n", ans);}else {scanf("%d", &v);if(v != 0) plus(1, a, b, v);}}return 0; }
解釋一下:query[]表示一個區間里面的和……然后左右半區間分別搞……1操作時Lazy優化是必須要加的(就是好像老師布置的作業等要交了再補……),然后,便過之。
問我為什么getval寫函數指針?因為這是百搭函數……要求最大值傳入max最小值傳min求和傳sum……
然后:就沒有然后了。過之。