Description
给出了一个序列,你需要处理如下两种询问。
"C a b c"表示给[a, b]区间中的值全部增加c (-10000 ≤ c ≤ 10000)。
"Q a b" 询问[a, b]区间中所有值的和。
Input
第一行包含两个整数N, Q。1 ≤ N,Q ≤ 100000.
第二行包含n个整数,表示初始的序列A (-1000000000 ≤ Ai ≤ 1000000000)。
接下来Q行询问,格式如题目描述。
Output
对于每一个Q开头的询问,你需要输出相应的答案,每个答案一行。
Sample Input
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
Sample Output
4
55
9
15
用树状数组的做法会超时,使用线段树进行区间更新
#include<stdio.h>
#include<string.h>
#define Min -10010
__int64 segtree[400000], a[100010], mark[400000];
void build(int l , int r, int id)
{
mark[id] = 0;
if(l == r)
{
segtree[id] = a[l];
return ;
}
int mid = (l+r)>>1;
build(l, mid, id<<1);
build(mid+1, r, id<<1|1);
segtree[id] = segtree[id<<1] + segtree[id<<1|1];
}
void Pushdown(int id, int l, int r)
{
if(mark[id])
{
mark[id<<1] += mark[id];
mark[id<<1|1] += mark[id];
int mid = (l+r)>>1;
segtree[id<<1] += (mark[id] * (mid - l + 1) );
segtree[id<<1|1] += ( mark[id] * (r - mid) );
mark[id] = 0;
}
}
__int64 query(int l, int r, int id, int L, int R)
{
if(L <= l && r <= R)
{
return segtree[id];
}
Pushdown(id, l, r);
int mid = (l+r)>>1;
__int64 sum = 0;
if(L <= mid)
sum += query(l, mid, id<<1, L, R);
if(R >= mid +1)
sum += query(mid+1, r, id<<1|1, L, R);
return sum;
}
void update(int l, int r, int id, int left, int right, int cnt)
{
if(left <= l && r <= right)
{
mark[id] += cnt;
segtree[id] += (cnt * (r-l+1) );
return ;
}
Pushdown(id, l, r);
int mid = (l+r)>>1;
if(left <= mid)
update(l, mid, id<<1, left, right, cnt);
if(right >= mid+1)
update(mid+1, r, id<<1|1, left, right, cnt);
segtree[id] = segtree[id<<1] + segtree[id<<1|1];
}
int main()
{
int i, j, k, n, m;
scanf("%d%d", &n, &m);
for(i = 1; i <= n; ++i)
scanf("%I64d", &a[i]);
build(1, n, 1);
char s[10];
for(i = 0; i < m; ++i)
{
scanf("%s", s);
if(strcmp(s, "Q") == 0)
{
int x, y;
scanf("%d%d", &x, &y);
printf("%I64d\n", query(1, n, 1, x, y) );
}
else
{
int x, y, z;
scanf("%d%d%d", &x, &y, &z);
update(1, n, 1, x, y, z);
}
}
return 0;
}