LibreOJ #6278. 数列分块入门 2 分块

https://loj.ac/problem/6278

思路：因为要询问区间内小于$x$的元素个数，$copy$原序列分块之后对每个块进行排序，这样在完整的块内就可以利用二分来快速得到答案。因为修改操作是对原序列进行的，对完整的块没有影响，但是对不完整的块是有影响的，这时候暴力修改后应该对该块重新$copy$并排序。且在查询不完整的块的时候，用的是原序列的值而不是$copy$的序列的值。

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

int a[50005];//原始序列
int b[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;	//可能会破坏单调性
		for(int i=l[posl];i<=r[posl];i++)
			b[i]=a[i];	//对 b 重新赋值并排序
		sort(b+l[posl],b+r[posl]+1);
	}
	else
	{
		for(int i=ll;i<=r[posl];i++)
			a[i]+=v;	//可能会破坏单调性
		for(int i=l[posl];i<=r[posl];i++)
			b[i]=a[i];	//对 b 重新赋值并排序
		sort(b+l[posl],b+r[posl]+1);
		for(int i=pos[ll]+1;i<=pos[rr]-1;i++)
			lazy[i]+=v;	//完整的块打上标记 不要修改
		for(int i=l[posr];i<=rr;i++)
			a[i]+=v;	//可能会破坏单调性
		for(int i=l[posr];i<=r[posr];i++)
			b[i]=a[i];	//对 b 重新赋值并排序
		sort(b+l[posr],b+r[posr]+1);
	}
}

int query(int ll,int rr,int v)
{
	int posl=pos[ll];
	int posr=pos[rr];
	int cnt=0;
	if(posl==posr)
	{
		for(int i=ll;i<=rr;i++)	//不完整的块 对原序列进行操作
			if(a[i]<v-lazy[posl])
				++cnt;
		return cnt;
	}
	for(int i=ll;i<=r[posl];i++)
		if(a[i]<v-lazy[posl])//不完整的块 对原序列进行操作
			++cnt;
	for(int i=l[posr];i<=rr;i++)
		if(a[i]<v-lazy[posr])//不完整的块 对原序列进行操作
			++cnt;
	for(int i=pos[ll]+1;i<=pos[rr]-1;i++)	//完整的块 对 b 序列进行操作
		cnt+=lower_bound(b+l[i],b+r[i]+1,v-lazy[i])-(b+l[i]);
	return cnt;
}

int main()
{
	//freopen("a1.in","r",stdin);
	//freopen("a2.out","w",stdout);
	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;
		r[i]=i*dis;
	}
	r[num]=n;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		b[i]=a[i];
		pos[i]=(i-1)/dis+1;
	}
	for(int i=1;i<=num;i++)
		sort(b+l[i],b+r[i]+1);
	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(t2,t3,t4*t4));
	}
	return 0;
}