P3374 【模板】树状数组 1（线段树版）

区间更新查询实现

原创于 2017-12-20 19:40:56 发布 · 92 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种使用线段树进行区间更新与查询的算法实现。该算法能够高效地处理一系列区间内的元素更新操作及查询指定位置元素的累计值。通过递归方式构建线段树，并实现了更新与查询功能。

#include<cstring>
int n,len=0;
struct nod{int l,r,ls,rs,c;nod(){c=0;};}a[1000010];
void br(int l,int r)
{
    len++;int x=len;
    a[x].l=l;a[x].r=r;
    int m=(l+r)/2;
    if(l!=r)    
    {
        a[x].ls=len+1;br(l,m);
        a[x].rs=len+1;br(m+1,r);
    }
}
void gd(int x,int k,int c)
{
    a[x].c+=c;
    if(a[x].l==a[x].r&&a[x].l==k)return ;
    int m=(a[x].l+a[x].r)/2;
    if(k<=m){gd(a[x].ls,k,c);}
    else{gd(a[x].rs,k,c);}
}
int cz(int x,int l,int r)
{
    if(a[x].l==l&&a[x].r==r){return a[x].c;}
    int m=(a[x].l+a[x].r)/2;
    if(r<=m){return cz(a[x].ls,l,r);}
    else if(l>=m+1){return cz(a[x].rs,l,r);}
    else{return (cz(a[x].ls,l,m)+cz(a[x].rs,m+1,r));}
}
int main()
{
    scanf("%d",&n);int k;scanf("%d",&k);
    br(1,n);
    for(int i=1;i<=n;i++)
    {
        int x;scanf("%d",&x);
        gd(1,i,x);
    }
    for(int i=1;i<=k;i++)
    {
        int tt,x,y,c;
        scanf("%d",&tt);scanf("%d %d",&x,&c);
        if(tt==1)
        {
            gd(1,x,c);
        }
        if(tt==2)
        {
            printf("%d\n",cz(1,x,c));
        }
    }
}