数列分块入门 1
链接:LibreOj6277
题目描述
给出一个长为n的数列,以及n个操作,操作涉及区间加法,单点查值。
输入格式
第一行输入一个数字n。
第二行输入n个数字,第i个数字为 a[i],以空格隔开。
接下来输入n行询问,每行输入四个数字op、l、r、c,以空格隔开。
若 op==0,将[l,r]之间的数字都加上c
若 op==1,表示询问a[r]的值( l 和 c 忽略)。
输出格式
对于每次询问,输出一行一个数字表示答案。
样例输入
4
1 2 2 3
0 1 3 1
1 0 1 0
0 1 2 2
1 0 2 0
样例输出
2
5
分析:
设置一个加法标记用于记录每一个块中元素每一个加了多少,每次操作对每个整块修改加法标记,
对不完整的块暴力修改
每次询问时输出元素的值加上其所在块的加法标记。
我的代码:
#include<iostream>
#include<cstdio>
#include<cmath>
typedef long long ll;
const int inf=0x3f3f3f3f;
const int inn=0x80808080;
using namespace std;
const int maxm=5e4+5;
int n;
int a[maxm];
int l[maxm],r[maxm];
int belong[maxm];
int add[maxm];
int block,num;
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;
add[i]=0;//初始化加法标记
}
r[num]=n;
for(int i=1;i<=n;i++){
belong[i]=(i-1)/block+1;
}
}
void update(int x,int y,int val){
if(belong[x]==belong[y]){
for(int i=x;i<=y;i++){//暴力修改
a[i]+=val;
}
return ;
}
for(int i=x;i<=r[belong[x]];i++){//暴力修改
a[i]+=val;
}
for(int i=l[belong[y]];i<=y;i++){//暴力修改
a[i]+=val;
}
for(int i=belong[x]+1;i<belong[y];i++){//修改x和y之间的块的加法标记
add[i]+=val;
}
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){//输入初始值
scanf("%d",&a[i]);
}
build();
for(int i=1;i<=n;i++){
int d,x,y,c;
scanf("%d%d%d%d",&d,&x,&y,&c);
if(d==1){
printf("%d\n",a[y]+add[belong[y]]);
}else{
update(x,y,c);
}
}
return 0;
}