【SSL2648】线段树练习五

Description

一行N个方格，开始每个格子里的数都是0。现在动态地提出一些问题和修改：提问的形式是求某一个特定的子区间[a,b]中所有元素的和；修改的规则是指定某一个格子x，加上或者减去一个特定的值A。现在要求你能对每个提问作出正确的回答。1≤N≤100000,提问和修改的总数可能达到100000条。

Input

20 //方格个数 
6 //有几组操作 
M 1 1 //表示修改,第一个表示格子位置,第二个数表示在原来的基础上加上的数, 
M 2 2 
M 3 4 
M 3 -5 
M 6 7 
C 2 6 //表示统计 ,第一个数表示起始位置,第二个数表示结束位置

Output

8

思路:
先建立一个线段树，维护$l-r$之间的$sum$值，在更改时找到要更改的那个点，直接更改它的值，在查询时找到包括在内的区间，直接加上它的值

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
int m, n;
struct node{
	int l, r, w;
}tree[100000<<2];
void build(int k, int l, int r){
	tree[k].l=l;tree[k].r=r;
	if(l==r)return;
    int mid=l+r>>1;
    build(k*2, l, mid);
    build(k*2+1, mid+1, r);
}
void change(int k, int x, int add){
	if(tree[k].l==x&&tree[k].r==x){
		tree[k].w+=add;
		return;
	}
	if(x<=tree[k*2].r)change(k*2, x, add);
	else change(k*2+1, x, add);
	tree[k].w=tree[k*2].w+tree[k*2+1].w;
}
int sum(int k, int l, int r){
	if(tree[k].l==l&&tree[k].r==r){
		return tree[k].w;
	}
	if(r<=tree[k*2].r)return sum(k*2, l, r);
	if(l>=tree[k*2+1].l)return sum(k*2+1, l, r);
	else return sum(k*2, l, tree[k*2].r)+sum(k*2+1, tree[k*2+1].l, r);
}
int main(){
	scanf("%d%d ", &m, &n);
	build(1, 1, m);
	for(int i=1; i<=n; i++){
		char c;
		int x, y;
		scanf("%c %d %d ",&c, &x, &y);
		if(c=='M'){
			change(1, x, y);
		}
		else{
			printf("%d\n",sum(1, x, y));
		}
	}
	return 0;
}