【题目描述】
这是一道模板题。
给定数列a[1],a[2],…,a[n] 你需要依次进行q个操作,操作有两类:
1,lrx :给定 l,r,x ,对于所有i∈[l,r],将a[i]加上x (换言之,将a[l],a[l+1],…,a[r]分别加上x);
2、lr :给定l,r,求∑ri=la[i]的值(换言之,求a[l]+a[l+1]+⋯+a[r]的值)。
【输入】
第一行包含2 个正整数n,q ,表示数列长度和询问个数。保证1≤n,q≤106 。
第二行n个整数a[1],a[2],…,a[n] ,表示初始数列。保证∣∣a[i]∣≤106 。
接下来q行,每行一个操作,为以下两种之一:
1、lrx :对于所有i∈[l,r],将a[i]加上x;
2、lr :输出∑ri=la[i]的值。
保证1≤l≤r≤n,∣x∣≤106 。
【输出】
对于每个2lr操作,输出一行,每行有一个整数,表示所求的结果。
【输入样例】
5 10
2 6 6 1 1
2 1 4
1 2 5 10
2 1 3
2 2 3
1 2 2 8
1 2 3 7
1 4 4 10
2 1 2
1 4 5 6
2 3 4
【输出样例】
15
34
32
33
50
【提示】
数据范围与提示:
对于所有数据,1≤n,q≤106,∣a[i]∣≤106,1≤l≤r≤n,∣x∣≤106 。
#include <bits/stdc++.h>
using namespace std;
long long bits_0[1000005], bits_1[1000005], n;
int l(int x) {
return x & (-x);
}
void U(long long *bits, int i, long long k) {
while (i <= n) {
bits[i] += k;
i += l(i);
}
}
long long P(long long *bits, int i) {
long long cnt = 0;
while (i > 0) {
cnt += bits[i];
i -= l(i);
}
return cnt;
}
long long R(int left, int right) {
long long ans = P(bits_0, right) * right + P(bits_1, right);
return ans - P(bits_0, left - 1) * (left - 1) - P(bits_1, left - 1);
}
int main() {
int q, delta, left, right;
char judge;
while (~scanf("%d%d", &n, &q)) {
memset(bits_0, 0, sizeof(bits_0));
memset(bits_1, 0, sizeof(bits_1));
for (int i = 1; i <= n; i++) {
scanf("%d", &delta);
U(bits_1, i, delta);
}
while (q--) {
cin>>judge;
if (judge == 'C') {
scanf("%d%d%d", &left, &right, &delta);
U(bits_0, left, delta);
U(bits_1, left, 1ll * -delta * (left - 1));
U(bits_0, right + 1, -delta);
U(bits_1, right + 1, 1ll * delta * right);
} else {
scanf("%d%d", &left, &right);
printf("%lld\n", R(left, right));
}
}
}
return 0;
}