洛谷P3372（线段树）

线段树模版洛谷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;
}