LibreOJ #6277. 数列分块入门 1 分块

https://loj.ac/problem/6277

思路：分块$+lazy$标记。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#define INF 0x3f3f3f3f
using namespace std;

int a[50005];//原始序列
int l[2000];//第i个块的左区间端点
int r[2000];//第i个块的右区间端点
int pos[50005];//第i个点属于第几个块
int lazy[2000];

void add(int ll,int rr,int v)
{
	int posl=pos[ll];//要修改的区间的左端点属于第几个块
	int posr=pos[rr];//要修改的区间的右端点属于第几个块
	if(posl==posr)//同一块中 暴力
	{
		for(int i=ll;i<=rr;i++)
			a[i]+=v;
	}
	else //不同块中
	{
		for(int i=ll;i<=r[posl];i++)//暴力修改左侧不完整的块
			a[i]+=v;
		for(int i=pos[ll]+1;i<=pos[rr]-1;i++)//修改中间完整的块 lazy标记
			lazy[i]+=v;
		for(int i=l[posr];i<=rr;i++)//暴力修改右侧不完整的块
			a[i]+=v;
	}
}

int query(int i)
{
	int posi=pos[i];
	if(lazy[posi])
		return a[i]+lazy[posi];
	else
		return a[i];
}

int main()
{
	int n;
	scanf("%d",&n);
	int dis=sqrt(n);//每一块的大小
	int num=ceil(n*1.0/dis);//分块数目
	for(int i=1;i<=num;i++)
	{
		l[i]=(i-1)*dis+1;//第i个块的左区间端点
		r[i]=i*dis;//第i个块的右区间端点
	}
	r[num]=n;//最后一个块的右端点最大等于n
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		pos[i]=(i-1)/dis+1;//第i个数属于第几个块
	}
	int t1,t2,t3,t4;
	for(int i=0;i<n;i++)
	{
		scanf("%d %d %d %d",&t1,&t2,&t3,&t4);
		if(t1==0)
			add(t2,t3,t4);
		else
			printf("%d\n",query(t3));
	}
	return 0;
}