码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2014-10-06 20:08:40      阅读:188      评论:0      收藏:0      [点我收藏+]

标签:des   style   blog   color   io   os   ar   for   strong   

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单点修改区间求和线段树

标签:des   style   blog   color   io   os   ar   for   strong   

原文地址:http://www.cnblogs.com/seekdreamer/p/4008432.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!