线段树(模板)(自拟)

由于我太弱了,所以现在才会线段树

#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;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值