线段树问题

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;
}  




  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值