Skip to main content

分块

参考资料

例题

洛谷 P3372 【模板】线段树 1

如题,已知一个数列 {ai}\{a_i\},你需要进行下面两种操作:

  1. 将某区间每一个数加上 kk
  2. 求出某区间每一个数的和。
Code (5)
#include <bits/stdc++.h>
#define s1 (u*3)
#define s2 (u*3+1)
#define s3 (u*3+2)
#define m1 (l+(r-l)/3)
#define m2 (r-(r-l)/3)
using namespace std;

using ll=long long;
const int N=100005;
ll a[N],val[N<<4],tag[N<<4];
void gx(int u,ll v,int len)
{
val[u]+=v*len;
tag[u]+=v;
}
void push_up(int u)
{
val[u]=val[s1]+val[s2]+val[s3];
}
void push_down(int u,int l,int r)
{
if(!tag[u])return;
if(l<=m1)gx(s1,tag[u],m1-l+1);
if(m1<m2)gx(s2,tag[u],m2-m1);
if(r>m2)gx(s3,tag[u],r-m2);
tag[u]=0;
}
void build(int u,int l,int r)
{
if(l==r){val[u]=a[l];return;}
if(l<=m1)build(s1,l,m1);
if(m1<m2)build(s2,m1+1,m2);
if(r>m2)build(s3,m2+1,r);
push_up(u);
}
void update(int u,int l,int r,int x,int y,ll v)
{
if(x>r||y<l)return;
if(x<=l&&r<=y){gx(u,v,r-l+1);return;}
push_down(u,l,r);
if(l<=m1)update(s1,l,m1,x,y,v);
if(m1<m2)update(s2,m1+1,m2,x,y,v);
if(r>m2)update(s3,m2+1,r,x,y,v);
push_up(u);
}
ll query(int u,int l,int r,int x,int y)
{
if(x>r||y<l)return 0;
if(x<=l&&r<=y)return val[u];
push_down(u,l,r);
ll res=0;
if(l<=m1)res+=query(s1,l,m1,x,y);
if(m1<m2)res+=query(s2,m1+1,m2,x,y);
if(r>m2)res+=query(s3,m2+1,r,x,y);
return res;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>a[i];
}
build(1,1,n);
while(m--)
{
int op,x,y;
ll k;
cin>>op;
if(op==1)
{
cin>>x>>y>>k;
update(1,1,n,x,y,k);
}
else if(op==2)
{
cin>>x>>y;
cout<<query(1,1,n,x,y)<<'\n';
}
}
return 0;
}

洛谷 P2801 教主的魔法

教主最近学会了一种神奇的魔法,能够使人长高。于是他准备演示给 XMYZ 信息组每个英雄看。于是 NN 个英雄们又一次聚集在了一起,这次他们排成了一列,被编号为 1,2,,N1, 2, \dots, N

每个人的身高一开始都是不超过 10001000 的正整数。教主的魔法每次可以把闭区间 [L,R][L, R]1LRN1≤L≤R≤N)内的英雄的身高全部加上一个整数 WW。(虽然 L=RL=R 时并不符合区间的书写规范,但我们可以认为是单独增加第 L(R)L(R) 个英雄的身高)

CYZ、光哥和 ZJQ 等人不信教主的邪,于是他们有时候会问 WD 闭区间 [L,R][L, R] 内有多少英雄身高大于等于 CC,以验证教主的魔法是否真的有效。

WD 巨懒,于是他把这个回答的任务交给了你。

Code (1)
#include <bits/stdc++.h>
using namespace std;

using ll=long long;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
const int N=1000005;
int a[N],b[N],t[N];
int id[N],st[N],ed[N];
void reset(int k)
{
for(int i=st[k];i<=ed[k];i++)t[i]=a[i];
sort(t+st[k],t+ed[k]+1);
}
void init(int n)
{
int len=sqrt(n);
for(int i=1;i<=len;i++)
{
st[i]=n/len*(i-1)+1;
ed[i]=n/len*i;
}
ed[len]=n;
for(int i=1;i<=len;i++)
{
for(int j=st[i];j<=ed[i];j++)id[j]=i;
}
for(int i=1;i<=len;i++)
{
reset(i);
}
}
void update(int l,int r,int v)
{
int x=id[l],y=id[r];
if(x==y)
{
for(int i=l;i<=r;i++)a[i]+=v;
reset(x);
return;
}
for(int i=l;i<=ed[x];i++)a[i]+=v;
for(int i=st[y];i<=r;i++)a[i]+=v;
for(int i=x+1;i<y;i++)b[i]+=v;
reset(x);
reset(y);
}
int query(int l,int r,int v)
{
int x=id[l],y=id[r];
int ans=0;
if(x==y)
{
for(int i=l;i<=r;i++)if(a[i]+b[x]>=v)ans++;
return ans;
}
for(int i=l;i<=ed[x];i++)if(a[i]+b[x]>=v)ans++;
for(int i=st[y];i<=r;i++)if(a[i]+b[y]>=v)ans++;
for(int i=x+1;i<=y-1;i++)ans+=ed[i]-(lower_bound(t+st[i],t+ed[i]+1,v-b[i])-t)+1;
return ans;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n,q;
cin>>n>>q;
for(int i=1;i<=n;i++)
{
cin>>a[i];
}
init(n);
while(q--)
{
char op;
int l,r,v;
cin>>op>>l>>r>>v;
if(op=='M')update(l,r,v);
else if(op=='A')cout<<query(l,r,v)<<'\n';
}
return 0;
}