单点赋值 求区间和

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
using namespace std;
#define MAXN 100005
typedef struct segmentree
{
    int       left;
    int       right;
    long long sum;
}segmentree;
segmentree segtree[MAXN<<2];
long long a[MAXN];
void push_up(int root)
{
    segtree[root].sum=segtree[root*2].sum+segtree[root*2+1].sum;
}
void CreateSegmentree(int root,int l,int r)  //创建线段树
{
    segtree[root].left = l;
    segtree[root].right = r;
    if(l == r)
    {
         segtree[root].sum=a[l];
    }
    else
    {
     int mid = (l+r)>>1;
     CreateSegmentree(root<<1,l,mid);
     CreateSegmentree(root<<1|1,mid+1,r);
     push_up(root);
    }
    return;
}
long long query(int root,int l,int r)
{
     if(segtree[root].left>r||segtree[root].right<l)
     return 0;
     else if(segtree[root].left==l&&segtree[root].right==r)
     return segtree[root].sum;
     int mid=(segtree[root].left + segtree[root].right) >> 1;
     if(l > mid)
     return query(root<<1|1,l,r);
     else if(r <= mid)
     return query(root<<1,l,r);
     else
     return query(root<<1,l,mid)+query(root<<1|1,mid + 1,r);
}
void revise(int root,int x, long long val)    //单点修改,求区间值
{
    if(segtree[root].left==segtree[root].right)
    {
        segtree[root].sum =val;
    }
    else
    {
        if(x<= segtree[root<<1].right)
        revise(root<<1,x,val);
        else
        revise(root<<1|1,x,val);
        push_up(root);
    }
    return ;
}
int main()
{
    int l,r;
    int x;
    long long val;
    int n,i,m;
    char ops[20];
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)
    {
       scanf("%lld",&a[i]);
    }
    CreateSegmentree(1,1,n);
    while(m--)
    {
            scanf("%s",ops);
            if(ops[0]=='1')
            {
                scanf("%d%d",&l,&r);
                printf("%lld\n",query(1,l,r));
            }
            else
            {
                scanf("%d%lld",&x,&val);
                revise(1,x,val);
            }
    }
  return 0;
}

猜你喜欢

转载自blog.csdn.net/xigongdali/article/details/82720192