分块是一种暴力,可以像线段树那样维护区间信息,复杂度不及线段树,但维护的东西不需要满足区间加法。
例题1:POJ3468
题意:有一个序列,有两种操作,一种是查询区间和,一种是区间内所有数值加上v,很明显的线段树裸题,但这题也可以用分块做。
分块和线段树一样,只不过一个序列只分一次块,不像线段树分到不能再细分,因为是区间更新,这里也需要laze标记。更新的时候,只有整个完整的块都被更新的才需要加laze标记,表示这个块内所有的数值都加上了laze[i]的值,对于不完整的块,直接暴力修改每一个a[i]即可。
#include<stdio.h>
using namespace std;
#include<math.h>
const int maxn = 3e5 + 10;
long long block,num,l[maxn],r[maxn],belong[maxn],block_sum[maxn];
long long n,q,a[maxn],sum[maxn],add[maxn];
void build() {
block = sqrt(n);
num = n / block;
if(n % block) num++;
for(int i = 1; i <= num; i++) {
l[i] = (i - 1) * block + 1;
r[i] = i * block;
}
r[num] = n;
for(int i = 1; i <= n; i++) {
belong[i] = (i - 1) / block + 1;
}
for(int i = 1; i <= num; i++)
block_sum[i] = sum[r[i]] - sum[l[i] - 1];
}
void update(int x,int y,long long v) {
int bx = belong[x],by = belong[y];
if(bx == by) {
for(int i = x; i <= y; i++)
a[i] += v;
block_sum[bx] += 1ll * (y - x + 1) * v;
return ;
}
for(int i = x; i <= r[bx]; i++)
a[i] += v;
for(int i = l[by]; i <= y; i++)
a[i] += v;
block_sum[by] += 1ll * (y - l[by] + 1) * v;
block_sum[bx] += 1ll * (r[bx] - x + 1) * v;
for(int i = bx + 1; i < by; i++) {
block_sum[i] += 1ll * (r[i] - l[i] + 1) * v;
add[i] += v;
}
}
long long query(int x,int y) {
long long ans = 0;
int bx = belong[x],by = belong[y];
if(bx == by) {
for(int i = x; i <= y; i++)
ans += a[i] + add[bx];
return ans;
}
for(int i = x; i <= r[bx]; i++)
ans += a[i] + add[bx];
for(int i = l[by]; i <= y; i++) {
ans += a[i] + add[by];
}
for(int i = bx + 1; i < by; i++) {
ans += block_sum[i];
}
return ans;
}
char op[10];
int main() {
scanf("%lld%lld",&n,&q);
for(int i = 1; i <= n; i++) {
scanf("%lld",&a[i]);
}
for(int i = 1; i <= n; i++) {
sum[i] = sum[i - 1] + a[i];
}
build();
for(int i = 1; i <= q; i++) {
scanf("%s",op);
int x,y,c;
if(op[0] == 'Q') {
scanf("%d%d",&x,&y);
printf("%lld\n",query(x,y));
}
else {
scanf("%d%d%d",&x,&y,&c);
update(x,y,c);
}
}
return 0;
}
例题2:洛谷p2801:教主的魔法
题目大意:教主最近学会了一种神奇的魔法,能够使人长高。于是他准备演示给XMYZ信息组每个英雄看。于是N个英雄们又一次聚集在了一起,这次他们排成了一列,被编号为1、2、……、N。
每个人的身高一开始都是不超过1000的正整数。教主的魔法每次可以把闭区间[L, R](1≤L≤R≤N)内的英雄的身高全部加上一个整数W。(虽然L=R时并不符合区间的书写规范,但我们可以认为是单独增加第L(R)个英雄的身高)
CYZ、光哥和ZJQ等人不信教主的邪,于是他们有时候会问WD闭区间 [L, R] 内有多少英雄身高大于等于C,以验证教主的魔法是否真的有效。
WD巨懒,于是他把这个回答的任务交给了你。
你可能会想到主席树来查询,但主席树做不了区间更新,因此主席树gg。这题还是要分块,每个块开一个桶,每个桶内的数据排序一下,查找时按块二分查找即可,复杂度为o(q * sqrt(n) * log(sqrt(n)))
更新时需要重新维护桶,中间完整的桶可以直接用tag标记,到时查找c - tag即可,两边不完整的块,要修改每个数的值,并重新排序。复杂度为o(q * sqrt(n))。
初始化复杂度为o(n * log(sqrt(n)))。
整体复杂度为:O(q * sqrt(n) * log(sqrt(n))),这题卡了常,要手写二分,加快读。开O2优化的话快5倍左右。
本题总结:分块着实是一种不错的数据结构,简单,有时使用数据结构不是一定要维护出什么值,而是为了方便处理,降低复杂度,必须改掉生搬硬套的惯性思维,要灵活运用。
#include<stdio.h>
#include<math.h>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 2e6 + 10;
vector<ll> g[1100];
ll num,block,belong[maxn],l[1100],r[1100],n,q,tag[1100];
ll a[maxn];
inline int read(){
ll s=0,w=1;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
return s*w;
}
void build() {
block = sqrt(n);
num = n / block;
if(n % block) num++;
for(int i = 1; i <= num; i++) {
l[i] = (i - 1) * block + 1;
r[i] = i * block;
}
r[num] = n;
for(int i = 1; i <= n; i++) {
belong[i] = (i - 1) / block + 1;
}
for(int i = 1; i <= n; i++) {
int pos = belong[i];
g[pos].push_back(a[i]);
}
for(int i = 1; i <= num; i++)
sort(g[i].begin(),g[i].end());
}
int lowerbound(int p,int x) {
int l = 0,r = g[p].size();
while(l < r) {
int mid = l + r >> 1;
if(g[p][mid] >= x) r = mid;
else l = mid + 1;
}
return g[p].size() - l;
}
void update(int li,int ri,ll c) {
int pl = belong[li],pr = belong[ri];
if(pl == pr) {
g[pl].clear();
for(int i = l[pl]; i < li; i++) g[pl].push_back(a[i]);
for(int i = li; i <= ri; i++) g[pl].push_back(a[i] += c);
for(int i = ri + 1; i <= r[pl]; i++) g[pl].push_back(a[i]);
sort(g[pl].begin(),g[pl].end());
return ;
}
g[pl].clear();
for(int i = l[pl]; i < li; i++) g[pl].push_back(a[i]);
for(int i = li; i <= r[pl]; i++) g[pl].push_back(a[i] += c);
sort(g[pl].begin(),g[pl].end());
g[pr].clear();
for(int i = l[pr]; i <= ri; i++) g[pr].push_back(a[i] += c);
for(int i = ri + 1; i <= r[pr]; i++) g[pr].push_back(a[i]);
sort(g[pr].begin(),g[pr].end());
for(int i = pl + 1; i < pr; i++)
tag[i] += c;
}
int query(int li,int ri,ll c) {
int pl = belong[li],pr = belong[ri];
int ans = 0;
if(pl == pr) {
int ans = 0;
for(int i = li; i <= ri; i++) {
if(a[i] + tag[pl] >= c) ans++;
}
return ans;
}
for(int i = li; i <= r[pl]; i++) {
if(a[i] + tag[pl] >= c) ans++;
}
for(int i = l[pr]; i <= ri; i++) {
if(a[i] + tag[pr] >= c) ans++;
}
for(int i = pl + 1; i < pr; i++) {
ans += lowerbound(i,c - tag[i]);
}
return ans;
}
char op[10];
int main() {
scanf("%d%d",&n,&q);
for(int i = 1; i <= n; i++) {
a[i] = read();
}
build();
for(int i = 1; i <= q; i++) {
scanf("%s",op);
ll x,y,z;
x = read();y = read();z = read();
if(op[0] == 'A') {
printf("%d\n",query(x,y,z));
}
else {
update(x,y,z);
}
}
return 0;
}