【板子】分块解决线段树问题

不得不说,分块思想是真的帅啊

P3372 【模板】线段树 1 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题意:

思路:

直接贴链接!

算法学习笔记(23): 分块 - 知乎 (zhihu.com)

Code:

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int mxn=2e5+10;

int n,m,sq,op,l,r,x;
int a[mxn],block_st[mxn],block_ed[mxn],bl[mxn],sum[mxn],add[mxn];
void modify_add(int l,int r,int x){
    if(bl[l]==bl[r]){
        for(int j=l;j<=r;j++){
            a[j]+=x;
            sum[bl[j]]+=x;
        }
    }else{
        for(int j=l;j<=block_ed[bl[l]];j++){
            a[j]+=x;
            sum[bl[j]]+=x;
        }
        for(int j=block_st[bl[r]];j<=r;j++){
            a[j]+=x;
            sum[bl[j]]+=x;
        }
        for(int i=bl[l]+1;i<bl[r];i++){
            add[i]+=x;
        }
    }
}
int query_sum(int l,int r){
    int res=0;
    if(bl[l]==bl[r]){
        for(int j=l;j<=r;j++) res+=a[j]+add[bl[j]];
    }else{
        for(int j=l;j<=block_ed[bl[l]];j++){
            res+=a[j]+add[bl[j]];
        }
        for(int j=block_st[bl[r]];j<=r;j++){
            res+=a[j]+add[bl[j]];
        }
        for(int i=bl[l]+1;i<bl[r];i++){
            res+=sum[i]+add[i]*(block_ed[i]-block_st[i]+1);
        }
    }
    return res;
}
void solve(){
    cin>>n>>m;
    sq=sqrt(n);
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=sq;i++){
        block_st[i]=n/sq*(i-1)+1;
        block_ed[i]=n/sq*i;
    }
    block_ed[sq]=n;
    for(int i=1;i<=sq;i++){
        for(int j=block_st[i];j<=block_ed[i];j++){
            sum[i]+=a[j];
            bl[j]=i;
        }
    }
    for(int i=1;i<=m;i++){
        cin>>op;
        if(op==1){
            cin>>l>>r>>x;
            modify_add(l,r,x);
        }else{
            cin>>l>>r;
            cout<<query_sum(l,r)<<'\n';
        }
    }
}
signed main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int __=1;//cin>>__;
    while(__--)solve();return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值