线段数模板

#include<cstdio>
using namespace std;
const int len=400001;
struct node{
    long long tag,data;
} tr[len];
int n,m,op,x,y;
long long ans,a[len],k;
void build_tree(int x,int l,int r){
    if (l==r){tr[x].tag=0;tr[x].data=a[l];return;}
    int mid=(l+r)/2;
    build_tree(x<<1,l,mid);
    build_tree((x<<1)+1,mid+1,r);
    tr[x].data=tr[x<<1].data+tr[(x<<1)+1].data; tr[x].tag=0;
}
void down(int x,int l,int r){
    int mid=(l+r)/2;
    tr[x<<1].tag+=tr[x].tag;
    tr[(x<<1)+1].tag+=tr[x].tag;
    tr[x<<1].data+=(long long)tr[x].tag*(mid-l+1); 
    tr[(x<<1)+1].data+=(long long)tr[x].tag*(r-mid);
    tr[x].tag=0;
}
void change(int x,int l,int r,int ll,int rr,long long k){
    if (l==ll&&r==rr){tr[x].data+=(long long)k*(r-l+1);tr[x].tag+=k;return;}
    int mid=(l+r)/2;
    down(x,l,r);
    if (rr<=mid){
        change(x<<1,l,mid,ll,rr,k);
    } else 
        if(ll>mid){change((x<<1)+1,mid+1,r,ll,rr,k);}
         else {change(x<<1,l,mid,ll,mid,k);change((x<<1)+1,mid+1,r,mid+1,rr,k);}
    tr[x].data=tr[x<<1].data+tr[(x<<1)+1].data;	
}
long long query(int x,int l,int r,int ll,int rr){
    if (l==ll&&rr==r) return tr[x].data;
    int mid=(l+r)/2;
    down(x,l,r);
    if (rr<=mid) return query(x<<1,l,mid,ll,rr);//在左边
     else if (ll>mid)return query((x<<1)+1,mid+1,r,ll,rr);//在右边
          else 
          {
      	long long ans=query(x<<1,l,mid,ll,mid)+query((x<<1)+1,mid+1,r,mid+1,rr);
      	return ans;
          }
}
int main(){
     scanf("%d%d",&n,&m);
     for (int i=1;i<=n;i++) scanf("%lld",&a[i]);
     build_tree(1,1,n);
     for (int i=1;i<=m;i++){
            scanf("%d%d%d",&op,&x,&y);
            if (op==1){
                scanf("%lld",&k);
                change(1,1,n,x,y,k);
            }else if(op==2){
                ans=query(1,1,n,x,y);
                printf("%lld\n",ans);
            }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/k42946/article/details/80779211
今日推荐