LOJ6278 数列分块入门 2 分块 排序 二分

题目链接

题意：
给出一个长为$n$的数列，以及$n$个操作，操作涉及区间加法，询问区间内小于某个值$x$的元素个数。

题解：
感觉要是树形数据结构的话可能要维护一个能区间加的主席树，我现在不会，不知道能不能做。

树形数据结构不好做考虑更暴力的分块。（人家都告诉你是分块题了啊）。分块的话，整块的肯定是打标记，两侧零散的肯定是暴力修改。但是这个要询问小于某个值的个数，那么还要维护出一个顺序来。我们的做法是开两个数组，一个记原来的数组，一个是每个块排序后的数组。拆开整块的时候要先下放标记，修改原数组，然后把要排序的数组的这个块的数对应改成原数组的数，然后重新排序。查询的话对于整块的就在排好序的块里二分，零散的就在原数组中暴力数。这样就做完了。

维护的标记的话有还是有一些细节要注意的，最后复杂度是$O(n\sqrt{n}logn)$，我这个题写的代码不是很优秀，跑的有点慢。

代码：

#include <bits/stdc++.h>
using namespace std;

int n,pos[100020],a[100010],sz,l[100010],r[100010],b[100020],book[100010];
int ll[100010],rr[100010];
inline int read()
{
	int x=0,f=1;
	char s=getchar();
	while(s>'9'||s<'0')
	{
		if(s=='-')
		f=-1;
		s=getchar();
	}
	while(s>='0'&&s<='9')
	{
		x=x*10+s-'0';
		s=getchar();
	}
	return x;
}
int main()
{
	n=read();
	sz=sqrt(n-1)+1;
	for(int i=1;i<=n;++i)
	{
		a[i]=read();
		b[i]=a[i];
	}
	for(int i=1;i<=sz;++i)
	{		
		for(int j=(i-1)*sz+1;j<=min(n,i*sz);++j)
		{
			pos[j]=i;
			l[j]=(i-1)*sz+1;
			r[j]=min(n,i*sz);
		}	
		ll[i]=(i-1)*sz+1;
		rr[i]=min(n,i*sz);
		sort(b+(i-1)*sz+1,b+min(n,i*sz)+1);
	}
	for(int q=1;q<=n;++q)
	{
		int opt=read(),x=read(),y=read(),z=read();
		if(opt==0)
		{
			for(int i=pos[x]+1;i<=pos[y]-1;++i)
			book[i]+=z;
			if(book[pos[x]])
			{
				for(int i=l[x];i<=r[x];++i)
				a[i]+=book[pos[x]];
				book[pos[x]]=0;
			}
			if(book[pos[y]])
			{
				for(int i=l[y];i<=r[y];++i)
				a[i]+=book[pos[y]];
				book[pos[y]]=0;
			}
			if(pos[x]!=pos[y])
			{
				for(int i=x;i<=r[x];++i)
				a[i]+=z;
				for(int i=l[y];i<=y;++i)
				a[i]+=z;
				for(int i=l[x];i<=r[x];++i)
				b[i]=a[i];
				sort(b+l[x],b+r[x]+1);
				for(int i=l[y];i<=r[y];++i)
				b[i]=a[i];
				sort(b+l[y],b+r[y]+1);
			}
			else
			{
				for(int i=x;i<=y;++i)
				a[i]+=z;
				for(int i=l[x];i<=r[x];++i)
				b[i]=a[i];
				sort(b+l[x],b+r[x]+1);
			}
		}
		else
		{
			z=z*z;
			int ans=0;
			for(int i=pos[x]+1;i<=pos[y]-1;++i)
			{
				int le=ll[i],ri=rr[i],mid,ji=ll[i]-1;
				while(le<=ri)
				{
					mid=(le+ri)/2;
					if(b[mid]+book[i]<z)
					{
						ji=mid;
						le=mid+1;
					}
					else
					ri=mid-1;
				}
				ans+=ji-ll[i]+1;
			}
			if(book[pos[x]])
			{
				for(int i=l[x];i<=r[x];++i)
				{
					a[i]+=book[pos[x]];
					b[i]=a[i];
				}
				book[pos[x]]=0;
				sort(b+l[x],b+r[x]+1);				
			}			
			if(book[pos[y]])
			{
				for(int i=l[y];i<=r[y];++i)
				{
					a[i]+=book[pos[y]];
					b[i]=a[i];
				}
				book[pos[y]]=0;
				sort(b+l[y],b+r[y]+1);				
			}
			if(pos[x]!=pos[y])
			{
				for(int i=x;i<=r[x];++i)
				{
					if(a[i]<z)
					++ans;
				}
				for(int i=l[y];i<=y;++i)
				{
					if(a[i]<z)
					++ans;
				}
			}
			else
			{
				for(int i=x;i<=y;++i)
				{
					if(a[i]<z)
					++ans;
				}
			}
			printf("%d\n",ans);
		}
	}
	return 0;
}