1080 线段树练习

题目描述 Description

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

输入描述 Input Description

输入文件第一行为一个整数N，接下来是n行n个整数，表示格子中原来的整数。接下一个正整数m，再接下来有m行，表示m个询问，第一个整数表示询问代号，询问代号1表示增加，后面的两个数x和A表示给位置X上的数值增加A，询问代号2表示区间求和，后面两个整数表示a和b，表示要求[a,b]之间的区间和。

输出描述 Output Description

共m行，每个整数

样例输入 Sample Input

6

4

5

6

2

1

3

4

1 3 5

2 1 4

1 1 9

2 2 6

样例输出 Sample Output

22

22

数据范围及提示 Data Size & Hint

1≤N≤100000， m≤10000 。

代码

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cmath>
using namespace std;
struct node
{
    int left,right,value;
};
node point[1000000];
int father[1000000];
int g;

void buildtree(int i,int left,int right) //建树
{
    father[i]=(int)floor(i)/2.0;
    point[i].left=left;
    point[i].right=right;
    point[i].value=0;
    if (left==right) {father[i]=i;}
    else
    {
    buildtree(i<<1, left, (int)floor( (right+left) / 2.0));
    buildtree((i<<1) + 1, (int)floor( (right+left) / 2.0) + 1, right);
    }
}
//----------------------------------------------------------------------------
void update(int x,int add,int left,int right,int now) //更改
{
    int mid=(left+right)>>1;
    point[now].value+=add;
    if ( (right-left)<1 ) {g=left; return;}
    if ( (left<=x) && (x<=mid) ) { update(x,add,left,mid,2*now); }
    if ( ((mid+1)<=x) && (x<=right) ) { update(x,add,mid+1,right,2*now+1); }
}
//----------------------------------------------------------------------------
int search(int x,int left,int right,int begin,int end) //查找
{
    if ( (left==begin)&&(right==end) )  {return point[x].value;}
    int mid=(begin+end)>>1;
    int ans=0;

    if ( (left>=begin)&&(right<=mid) ) {ans+=search(2*x,left,right,begin,mid);}  //左边
    else if ( (left>=mid+1)&&(right<=end) ) {ans+=search(2*x+1,left,right,mid+1,end);}  //右边
    else {ans+=search(2*x,left,mid,begin,mid);
          ans+=search(2*x+1,mid+1,right,mid+1,end);}//中间
    return ans;
}
//----------------------------------------------------------------------------

int main()
{
    int n,m,k;
    cin>>n;
    buildtree(1,1,n);
    for (int i=1;i<=n;i++)
    {
        int ad;
        cin>>ad;
        update(i,ad,1,n,1);
    }
    cin>>m;
    for (int i=1;i<=m;i++)
    {
        int tmp;
        cin>>tmp;
        if (tmp==1)
        {
            int x,ad;
            cin>>x>>ad;
            update(x,ad,1,n,1);
        }
        if (tmp==2)
        {
            int p,q;
            cin>>p>>q;
            cout<<search(1,p,q,1,n)<<endl;
        }
    }
    return 0;
}