线段树模版洛谷3372
进行区间 修改数
查询区间和
#include <iostream>
#include<algorithm>
#include <stdio.h>
#include <string>
#include <string.h>
#include <map>
#include <math.h>
#include <vector>
#include <set>
#include <queue>
#include <map>
#include <deque>
//#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+7;
ll gcd(ll a,ll b)
{
return b?gcd(b,a%b):a;
}
priority_queue<int,vector<int>,greater<int> >q;
ll a[N],lazy[N],tr[N];
void f(ll p,ll l,ll r,ll k)
{
lazy[p]+=k;
tr[p]+=(r-l+1)*k;
}
void pushdown(ll i,ll l,ll r)
{
ll mid=(l+r)/2;
f(2*i,l,mid,lazy[i]);
f(2*i+1,mid+1,r,lazy[i]);
lazy[i]=0;
}
void pushup(ll i)
{
tr[i]=tr[i*2]+tr[i*2+1];
}
void build(ll i,ll l,ll r)//建树
{
lazy[i]=0;
if(l==r)
{
tr[i]=a[l];
return ;
}
ll mid=(l+r)>>1;
build(i*2,l,mid);
build(i*2+1,mid+1,r);
pushup(i);
}
void update(ll i,ll l,ll r,ll x,ll y,ll t)//[l,r]区间加t
{
if(l>=x&&y>=r)
{
tr[i]+=t*(r-l+1);
lazy[i]+=t;
return ;
}
pushdown(i,l,r);
ll mid=(l+r)>>1;
if(x<=mid)
update(i*2,l,mid,x,y,t);
if(y>mid)
update(i*2+1,mid+1,r,x,y,t);
pushup(i);
}
ll query(ll i,ll l,ll r,ll x,ll y)输出区间[l,r]的和
{
if(l>=x&&y>=r)
{
return tr[i];
}
pushdown(i,l,r);
ll ans=0;
ll mid=(l+r)>>1;
if(x<=mid)
{
ans+=query(i*2,l,mid,x,y);
}
if(y>mid)
{
ans+=query(i*2+1,mid+1,r,x,y);
}
return ans;
}
int main()
{
ll n,m;
cin>>n>>m;
for(ll i=1;i<=n;i++)
cin>>a[i];
build(1,1,n);
while(m--)
{
ll x;
cin>>x;
if(x==1)
{
ll b,c,d;
cin>>b>>c>>d;
update(1,1,n,b,c,d);
} else{
ll b,c;
cin>>b>>c;
cout<<query(1,1,n,b,c)<<endl;
}
}
return 0;
}