LibreOJ - 6277 數列分塊入門 1(分塊模板)
阿新 • • 發佈:2020-07-12
給出一個長為 nn 的數列,以及 nn 個操作,操作涉及區間加法,單點查值。
Input
第一行輸入一個數字 nn。
第二行輸入 nn 個數字,第 ii 個數字為 aiai,以空格隔開。
接下來輸入 nn 行詢問,每行輸入四個數字 optopt、ll、rr、cc,以空格隔開。
若 opt=0opt=0,表示將位於 [l,r][l,r] 的之間的數字都加 cc。
若 opt=1opt=1,表示詢問 arar 的值(ll 和 cc 忽略)。
Output
對於每次詢問,輸出一行一個數字表示答案。
Example
樣例輸入
4
1 2 2 3
0 1 3 1
1 0 1 0
0 1 2 2
1 0 2 0
樣例輸出
2
5
#include <bits/stdc++.h> using namespace std; long long a[50005], sum[50005],add[50005]; int L[50005], R[50005]; int pos[50005]; int n, t; void change(int l, int r, long long d) { int p = pos[l], q = pos[r]; if(p == q) { for(int i = l; i <=r; i++) a[i] += d; sum[p] += d * (r - l + 1); } else { for(int i = p + 1; i <= q - 1; i++) add[i] += d; for(int i = l; i <= R[p]; i++) a[i] += d; sum[p] += d * (R[p] - l + 1); for(int i = L[q]; i <= r; i++) a[i] += d; sum[q] += d * (r - L[q] + 1); } } long long ask(int l, int r) {int p = pos[l], q = pos[r]; long long ans = 0; if(p == q) { for(int i = l; i <= r; i++) ans += a[i]; ans += add[p] * (r - l + 1); } else { for(int i = p + 1; i <= q - 1; i++) ans += sum[i] + add[i] * (R[i] - L[i] + 1); for(int i = l; i <= R[p]; i++) ans += a[i]; ans += add[p] * (R[p] - l + 1); for(int i = L[q]; i <= r; i++) ans += a[i]; ans += add[q] * (r - L[q] + 1); } return ans; } int main() { cin >> n; for(int i = 1; i <= n; i++) { scanf("%lld", &a[i]); } t = sqrt(n); for(int i = 1; i <= t; i++) { L[i] = (i - 1) * sqrt(n) + 1; R[i] = i * sqrt(n); } if(R[t] < n) t++, L[t] = R[t - 1] + 1,R[t] = n; for(int i = 1; i <= t; i++)//每一塊 { for(int j = L[i]; j <= R[i]; j++) { pos[j] = i; sum[i] += a[j]; } } for(int i = 1; i <= n; i++) { int opt, l, r; long long c; scanf("%d%d%d%lld", &opt, &l, &r, &c); if(opt == 0) { change(l, r, c); } else { cout << ask(r, r) << endl; } } }