線段樹(模板)(自擬)

由於我太弱了,所以現在纔會線段樹

#include <iostream>
#include <cstdio>
#include <cmath>
using namespace std;
long long a[1000005],t[2000055],ans[2000005];
long long n,m;
long long make_l(long long x)
{
    return x<<1;
}
long long make_r(long long x)
{
    return x<<1|1;
}
long long push_sum(long long x)
{
    ans[x]=ans[make_l(x)]+ans[make_r(x)];
}
void build_tree(long long x,long long l,long long r)
{
	t[x]=0;
    if(l==r)
    {
        ans[x]=a[l];
        return;
    }
    long long mid=(l+r)>>1;
    build_tree(make_l(x),l,mid);
    build_tree(make_r(x),mid+1,r);
    push_sum(x);
}
void f(long long p,long long l,long long r,long long num)
{
    t[p]+=num;
    ans[p]+=num*(r-l+1);
}
void down(long long x,long long l,long long r)
{
    long long mid=(l+r)>>1;
    f(make_l(x),l,mid,t[x]);
    f(make_r(x),mid+1,r,t[x]);
    t[x]=0;
}
void update(long long al,long long ar,long long l,long long r,long long p,long long k)
{
    if(al<=l&&r<=ar)
    {
        ans[p]+=k*(r-l+1);
        t[p]+=k;
        return;
    }
    down(p,l,r);
    long long mid=(l+r)>>1;
    if(al<=mid)
    {
        update(al,ar,l,mid,make_l(p),k);
    }
    if(ar>mid)
    {
        update(al,ar,mid+1,r,make_r(p),k);
    }
    push_sum(p);
}
long long find(long long x,long long y,long long l,long long r,long long p)
   {
       long long sum=0;
       if(x<=l&&r<=y)
           return ans[p];
       long long mid=(l+r)>>1;
       down(p,l,r);
       if(x<=mid)
       sum+=find(x,y,l,mid,make_l(p));
       if(y>mid)
       sum+=find(x,y,mid+1,r,make_r(p));
       return sum;
   }
int main(){
   cin>>n>>m;
   for(int i=1;i<=n;i++)
   {
       scanf("%lld",&a[i]); 
   }
   build_tree(1,1,n);
   //cout<<"1";
   while(m--)
   {
       long long  wzx;
       scanf("%lld",&wzx);
       if(wzx==1)
       {
           long long b,c,d;
           scanf("%lld%lld%lld",&b,&c,&d);
           update(b,c,1,n,1,d);
       }
       if(wzx==2)
       {
           long long b,c;
           scanf("%lld%lld",&b,&c);
           printf("%lld\n",find(b,c,1,n,1));
       }
   }
   return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章