P1438 无聊的数列 36pts TLE求调

学术版

Dream__Sky @ 2023-07-10 21:46:25

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 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()
{
    ios::sync_with_stdio(0);
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=n;i++) b[i]=a[i]-a[i-1];

    build(1,1,n);
    for(int i=1,opt,l,r,k,d;i<=m;i++)
    {
        cin>>opt;
        if(opt==1)
        {
            cin>>l>>r>>k>>d;
            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 
        {
            cin>>k;
            cout<<ask(1,1,k)<<"\n";
        }
    }
    return 0;
}

by intawl @ 2023-07-10 21:48:09

把 cin 改成 scanf 就好了吧

宣传


|