给定一个长度为 N 的数列 A,以及 M 条指令,每条指令可能是以下两种之一:
C l r d
,表示把 A[l],A[l+1],…,A[r] 都加上 d。Q l r
,表示询问数列中第 l∼r 个数的和。
对于每个询问,输出一个整数表示答案。
输入格式
第一行两个整数 N,M。
第二行 N 个整数 A[i]。
接下来 M 行表示 M 条指令,每条指令的格式如题目描述所示。
输出格式
对于每个询问,输出一个整数表示答案。
每个答案占一行。
数据范围
1≤N,M≤105
|d|≤10000
|A[i]|≤109
输入样例:
10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4
输出样例:
4
55
9
15
解析:
#include <iostream>
#include <string>
#include <cstring>
#include <cmath>
#include <ctime>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <vector>
#include <set>
#include <math.h>
#include <map>
#include <sstream>
#include <deque>
#include <unordered_map>
#include <unordered_set>
#include <bitset>
#include <stdio.h>
#include <tuple>
using namespace std;
/*
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
*/
typedef long long LL;
// #define int long long
#define ld long double
//#define INT __int128
const LL INF = 0x3f3f3f3f3f3f3f3f;
typedef unsigned long long ULL;
typedef pair<long long, long long> PLL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
const int inf = 0x3f3f3f3f;
const LL mod = 998244353;
const ld eps = 1e-12;
const int N = 1e5 + 10, M = 350 + 10;
int n, m;
int A[N];
LL sum[M], add[M];
int len;
int get(int i) {
return i / len;
}
void modify(int l, int r, int d) {
if (get(l) == get(r)) {
for (int i = l; i <= r; i++) {
A[i] += d;
sum[get(i)] += d;
}
}
else {
int i = l, j = r;
while (get(i) == get(l))A[i] += d, sum[get(i)] += d, i++;
while (get(j) == get(r))A[j] += d, sum[get(j)] += d, j--;
for (int k = get(i); k <= get(j); k++) {
sum[k] += d * len;
add[k] += d;
}
}
}
LL ask(int l, int r) {
LL ret = 0;
if (get(l) == get(r)) {
for (int i = l; i <= r; i++) {
ret += A[i] + add[get(i)];
}
}
else {
int i = l, j = r;
while (get(i) == get(l))ret += A[i] + add[get(i)], i++;
while (get(j) == get(r))ret += A[j] + add[get(j)], j--;
for (int k = get(i); k <= get(j); k++) {
ret += sum[k];
}
}
return ret;
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
cin >> n >> m;
//scanf("%d%d", &n, &m);
len = sqrt(n);
for (int i = 1; i <= n; i++) {
cin >> A[i];
sum[get(i)] += A[i];
}
char op[2];
int l, r, d;
while (m--) {
cin >> op;
//scanf("%s", op);
if (op[0] == 'C') {
cin >> l >> r >> d;
//scanf("%d%d%d", &l, &r, &d);
modify(l, r, d);
}
else {
cin >> l >> r;
//scanf("%d%d", &l, &r);
LL ret = ask(l, r);
//printf("%lld\n", ret);
cout << ret << endl;
}
}
return 0;
}