SSL_2648 线段树练习题五

题意

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

思路

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

代码

#include<cstdio>
#include<iostream>
struct Ilovehuangzhenbang{
    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();
}