分块80pts求调

P3372 【模板】线段树 1

114514xxx @ 2024-08-13 15:53:36

#include<bits/stdc++.h>
#define int long long
const int N=5e6+25;
int a[N],sum[N],add[N];
int L[N],R[N];
int pos[N],n,m,k,x,y,opt;
using namespace std;
void add1(int l,int r,int d){
    int p=pos[l],q=pos[r];
    if(p==q){
        for(int i=l;i<=r;i++)
        a[i]+d;
        sum[p]+=(r-l+1)*d;
    }else{
        for(int i=p+1;i<=q-1;i++)add[i]+=d;
        for(int i=l;i<=R[p];i++)a[i]+=d;
        sum[p]+=d*(R[p]-l+1);
        for(int i=L[q];i<=r;i++)a[i]+=d;
        sum[q]+=d*(r-L[q]+1);
    }
}
int query(int l,int r){
    int ans=0;
    int p=pos[l],q=pos[r];
    if(p==q){
        for(int i=l;i<=r;i++)
            ans+=a[i];
        ans+=add[p]*(r-l+1);
    }else{
        for(int i=p+1;i<=q-1;i++)
            ans+=sum[i]+add[i]*(R[i]-L[i]+1);
        for(int i=l;i<=R[p];i++)ans+=a[i];
        ans+=add[p]*(R[p]-l+1);
        for(int i=L[q];i<=r;i++)ans+=a[i];
        ans+=add[q]*(r-L[q]+1);
    }
    return ans;
}
signed main(){
    std::ios::sync_with_stdio(0);
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    int t=sqrt(n);
    for(int i=1;i<=t;i++){
        L[i]=(i-1)*t+1;
        R[i]=i*t;
    }
    if(R[t]<n)t++,L[t]=R[t-1]+1,R[t]=n;
    for(int i=1;i<=t;i++){
        for(int j=L[i];j<=R[i];j++){
            pos[j]=i;
            sum[i]+=a[j];
        }
    }
    for(int i=1;i<=m;i++){
        cin>>opt>>x>>y;
        if(opt==1){
            cin>>k;
            add1(x,y,k);
        }else
            cout<<query(x,y)<<endl;
    }
    return 0;

}

by 114514xxx @ 2024-08-13 16:23:30

已AC,此贴完结。


|