【线段树】SSL_2648 线段树练习题五

题意

给出几个刚开始为0的格子，我们可以进行两种操作：
1、修改：M x y（给格子x加上y）
2、查询：C x y（格子x到格子y上的数的总和）

思路

线段树。用cover表示当前这个区间里的格子的总和，每次修改的时候我们查询到代表它那个范围的节点，然后递归回它的根节累加，查询就是查到了代表x到y的那个区间就返回cover，否则我们判断然后向它的子节点查找累加(这个方法慢了点，应该有一个更好的算法）。

代码

#include<cstdio>
#include<iostream>
struct node{
	int l,r,cover;
}tree[400001];
int n,m,x,y,d;
char c;
int read()
{
    int d=1,f=0;char c;
    while((c=getchar())<48||c>57) if(c=='-') d=-1;f=(f<<3)+(f<<1)+c-48;
    while((c=getchar())>=48&&c<=57) f=(f<<3)+(f<<1)+c-48;
    return d*f;
}
void build(int p,int l,int r)
{
	tree[p].l=l;tree[p].r=r;
	if (r-l>1)
	{
		int m=(l+r)>>1;
		build(2*p,l,m);
		build(2*p+1,m,r);
	}
}
void find(int p,int l,int r)
{
	int m=(tree[p].l+tree[p].r)>>1;
	if (tree[p].l==l&&tree[p].r==r) d=p;
	else if (r<=m) find(2*p,l,r);
	else if (l>=m) find(2*p+1,l,r);
	else if (tree[p].r-tree[p].l>1)
	{
		find(2*p,l,m);
		find(2*p+1,m,r);
	}
}
void insert(int p,int v)
{
	while (p>0)//一直给它的父节点更新
	{
		tree[p].cover+=v;
		p/=2;
	}
}
int count(int p,int l,int r)
{
	int m=(tree[p].l+tree[p].r)>>1;
	if (tree[p].l==l&&tree[p].r==r) return tree[p].cover;
	else if (r<=m) return count(2*p,l,r);
	else if (l>=m) return count(2*p+1,l,r);
	else if (tree[p].r-tree[p].l>1) return count(2*p,l,m)+count(2*p+1,m,r);
}
void init()
{
	scanf("%d%d",&n,&m);
	n++;//因为我建的树的叶节点是(1,2)类似这样的，所以当我们给第n个格子进行加的操作时会超范围，所以给n++
	build(1,1,n);
	for (int i=1;i<=m;i++)
	{
		std::cin>>c;
		x=read();y=read();//用了输入优化会快一点
		if (c=='M') 
		{
			find(1,x,x+1);//查询代表这个区间的节点
			insert(d,y);//从这个节点向它的父节点一直更新修改过后的值
		}
		else printf("%d\n",count(1,x,y+1));//和n++的原因类似
	}
}
int main()
{
	init();
}