P1438 36pts tle线段树求调

rt. 题目

#include <bits/stdc++.h>
#define int long long 
using namespace std;
int a[100001],b[100001],n,m;
struct info
{
    int l,r,tag,sum;
}t[400001];
inline int read() {
    int x = 0, m = 1;
    char ch = getchar();
    while(!isdigit(ch)) {
        if(ch == '-') m = -1;
        ch = getchar();
    }
    while(isdigit(ch)) {
        x = x * 10 + ch - 48;
        ch = getchar();
    }
    return x * m;
}
inline void write(int x) {
    if(x < 0) {
        putchar('-');
        write(-x);
        return;
    }
    if(x >= 10) write(x / 10);
    putchar(x % 10 + '0');
}

inline void build(int p,int l,int r)
{
    t[p].l=l,t[p].r=r;
    if(l==r) 
    {
        t[p].sum=b[l];
        return ;
    }
    int mid=(l+r)>>1;
    build(p*2,l,mid);
    build(p*2+1,mid+1,r);
    t[p].sum=t[p*2].sum+t[p*2+1].sum;
}
inline void spread(int p)
{
    if(t[p].tag)
    {
        t[p*2].sum+=t[p].tag*(t[p*2].r-t[p*2].l+1);
        t[p*2+1].sum+=t[p].tag*(t[p*2+1].r-t[p*2+1].l+1);

        t[p*2].tag+=t[p].tag;
        t[p*2+1].tag+=t[p].tag;

        t[p].tag=0;
    } 
}
inline void query(int p,int l,int r,int k)
{
    if(t[p].l>=l&&t[p].r<=r) 
    {
        t[p].tag+=k;
        t[p].sum+=(t[p].r-t[p].l+1)*k;
        return;
    }
    spread(p);
    int mid=(t[p].l+t[p].r)>>1;
    if(l<=mid) query(p*2,l,r,k);
    if(mid<r) query(p*2+1,l,r,k); 
    t[p].sum=t[p*2].sum+t[p*2+1].sum;
}
inline void add(int p,int l,int k)
{
    if(t[p].l==t[p].r)
    {
        t[p].tag+=k;
        t[p].sum+=k;
        return ;
    }
    spread(p);
    int mid=(t[p].l+t[p].r)>>1;
    if(l<=mid) add(p*2,l,k);
    else add(p*2+1,l,k); 
    t[p].sum=t[p*2].sum+t[p*2+1].sum;
}
inline int ask(int p,int l,int r)
{
    if(t[p].l==t[p].r)
        return t[p].sum;
    spread(p);
    int mid=(t[p].l+t[p].r)>>1,ans=0;
    if(mid>=l) ans+=ask(p*2,l,r);
    if(mid<r) ans+=ask(p*2+1,l,r);
    return ans;
}
signed main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++) a[i]=read();
    for(int i=1;i<=n;i++) b[i]=a[i]-a[i-1];

    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        int opt=read();
        if(opt==1)
        {
            int l=read(),r=read(),k=read(),d=read();
            add(1,l,k);
            if(l+1<=r) query(1,l+1,r,d);
            if(r<n) add(1,r+1,-(k+(r-l)*d));
        }
        else 
        {
            int k=read();
            write(ask(1,1,k));
            putchar('\n');
        }
    }
    return 0;
}

谢谢