【YbtOJ 线段树 - 1】求区间和

求区间和

---

题目

给定一数列，规定有两种操作，一是修改某个元素，二是求区间的连续和

---

输入

输出

对于每个询问输出对应的答案

---

输入样例

10 20
0 1 10
1 1 4
0 6 6
1 4 10
1 8 9
1 4 9
0 10 2
1 1 8
0 2 10
1 3 9
0 7 8
0 3 10
0 1 1
1 3 8
1 6 9
0 5 5
1 1 8
0 4 2
1 2 8
0 1 1

输出样例

10
6
0
6
16
6
24
14
50
41

---

解题思路

我们这题用的是线段树，没啥好说的，模板

---

程序如下

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>

using namespace std;

int n, m, k, a, b;

long long sum[1000001];

long long query(int k, int l, int r, int x, int y)
{
	if(x <= l && r <= y) return sum[k];
	int mid = (l + r) / 2;
	long long res = 0;
	if(x <= mid) res += query(k * 2, l, mid, x, y);
	if(y > mid) res += query(k * 2 + 1, mid + 1, r, x, y);
	return res;
}

void change(int k, int l, int r, int x, int v)
{
	if(l == r)
	{
		sum[k] += v;
		return;
	}
	int mid = (l + r) / 2;
	long long res = 0;
	if(x <= mid) change(k * 2, l, mid, x, v);
	else change(k * 2 + 1, mid + 1, r, x, v);
	sum[k] = sum[k * 2] + sum[k * 2 + 1];
} 

int main()
{
	scanf("%d%d",&n,&m);
	for(int i = 1; i <= m; ++i)
	{
		scanf("%d%d%d",&k,&a,&b);
		if(k == 0)
		{
			change(1, 1, n, a, b);
		}
		else if(k == 1)
		{
			long long ans = query(1, 1, n, a, b);
			printf("%lld\n",ans); 
		}
	}
	return 0;
}