首页 > 代码库 > 【线段树】线段树系列 0.2单点修改区间求和线段树

【线段树】线段树系列 0.2单点修改区间求和线段树

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

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;}

 

【线段树】线段树系列 0.2单点修改区间求和线段树