这是这几天才学的一个新的数据结构,因为时间问题,本人还没有进行实际操作过,所以这里只能给大家带来黄学长的大佬级代码,反正我一看就懂(蒟蒻还加了一些注释):
题目传送门: P2801 教主的魔法
大佬的代码:
//研究一发黄学长的代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
int n,q,m,block;//n是题目中的,q也是,m是总块数,block是块的大小
int a[1000001],b[1000001],pos[1000001],add[1000001];//a是原始数组,b是分块中的数组,pos是每一个人分别在哪个块中记录,add是块的标记
void reset(int x)//分块初始化操作 ,修改不完整块也要用这个操作,方便后面的二分查找
{
int l=(x-1)*block+1,r=min(x*block,n);//查找右边界
for(int i=l;i<=r;i++)
b[i]=a[i];
sort(b+l,b+r+1);
}
int find(int x,int v)//查询操作,二分
{
int l=(x-1)*block+1,r=min(x*block,n);
int last=r;
while(l<=r)//二分一下,完美结束!
{
int mid=(l+r)/2;
if(b[mid]<v)l=mid+1;
else r=mid-1;
}
return last-l+1;
}
void update(int x,int y,int v)//修改操作
{
if(pos[x]==pos[y])//如果修改都在同一个块中 ,暴力修改
{
for(int i=x;i<=y;i++)a[i]=a[i]+v;
}
else
{
for(int i=x;i<=pos[x]*block;i++)a[i]=a[i]+v;//修改x及所在块的后面,
for(int i=(pos[y]-1)*block+1;i<=y;i++)a[i]=a[i]+v;//修改y及其所在块的前面
}
reset(pos[x]);reset(pos[y]);
for(int i=pos[x]+1;i<pos[y];i++)
add[i]+=v;
}
int query(int x,int y,int v)//总的查询操作
{
int sum=0;
if(pos[x]==pos[y])//如果在同一个块中,暴力!!!
{
for(int i=x;i<=y;i++)if(a[i]+add[pos[i]]>=v)sum++;
}
else
{
for(int i=x;i<=pos[x]*block;i++)//处理x所在的残块,暴力!!!
if(a[i]+add[pos[i]]>=v)sum++;
for(int i=(pos[y]-1)*block+1;i<=y;i++)//处理y所在的残块,暴力!!!
if(a[i]+add[pos[i]]>=v)sum++;
}
for(int i=pos[x]+1;i<pos[y];i++)//处理x块和y块中间的完整的块。
sum+=find(i,v-add[i]);
return sum;
}
int main()
{
scanf("%d%d",&n,&q);
block=int(sqrt(n));
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
pos[i]=(i-1)/block+1;//这句话的意思是就是查找第i个英雄在第几个块
b[i]=a[i];
}
if(n%block)m=n/block+1;//有几个块,不为零说明存在不完整的块,所以说要+1
else m=n/block;
for(int i=1;i<=m;i++)reset(i);//初始化块
for(int i=1;i<=q;i++)
{
char ch[5];int x,y,v;
scanf("%s%d%d%d",ch,&x,&y,&v);
if(ch[0]=='M')update(x,y,v);
else printf("%d
",query(x,y,v));
}
return 0;
}
注意:
经过本人亲自打了一遍,我发现易错的地方有如下两点:
1、查询残块的时候一定不要把左右区间弄反了。
2、就是在分配第几个人在第几个块的时候一定要把编号-1!!!!