您的位置:首页 > 其它

[bzoj 3343]教主的魔法

2018-03-03 15:53 323 查看
教主最近学会了一种神奇的魔法,能够使人长高。于是他准备演示给XMYZ信息组每个英雄看。于是N个英雄们又一次聚集在了一起,这次他们排成了一列,被编号为1、2、……、N。

每个人的身高一开始都是不超过1000的正整数。教主的魔法每次可以把闭区间[L,R](1≤L≤R≤N)内的英雄的身高全部加上一个整数W。(虽然L=R时并不符合区间的书写规范,但我们可以认为是单独增加第L(R)个英雄的身高)

CYZ、光哥和ZJQ等人不信教主的邪,于是他们有时候会问WD闭区间 [L, R] 内有多少英雄身高大于等于C,以验证教主的魔法是否真的有效。

WD巨懒,于是他把这个回答的任务交给了你。

这道题是一道不错的分块题。首先单考虑询问,由于是问是否大于等于某值,可以先将各个块给排序,然后二分,头尾两个不完整块要暴力枚举,之后修改,给整块打上加法标记,不完整块要暴力修改,由于这样可能会破坏不完整块所在的块的单调性,所以要重新排序。那么这道题就解决了。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
int n,m;
int a[1100000],b[1100000],belong[1100000],bl[1100000],br[1100000],add[1100000];
void ps(int x)
{
int l=bl[x],r=br[x];
for(int i=l;i<=r;i++)b[i]=a[i];
sort(b+l,b+r+1);
}
void fk()
{
int cnt=2*sqrt(n);
for(int i=1;i<=n;i++)
{
int bg=(i-1)/cnt+1;
belong[i]=bg;
if(bl[bg]==0)bl[bg]=i,br[bg-1]=i-1;
}
br[belong
]=n;
}
int ef(int x,int w)
{
int l=bl[x],r=br[x],ans=bl[x]-1;
while(l<=r)
{
int mid=(l+r)/2;
if(b[mid]<w)
{
l=mid+1;
ans=mid;
}
else r=mid-1;
}
return br[x]-ans;
}
void solve1(int l,int r,int c)
{
int lg=belong[l],rg=belong[r];
if(lg==rg)
{
for(int i=l;i<=r;i++)a[i]+=c;
ps(lg);
return ;
}
for(int i=l;i<=br[lg];i++)a[i]+=c;ps(lg);
for(int i=bl[rg];i<=r;i++)a[i]+=c;ps(rg);
for(int i=lg+1;i<=rg-1;i++)add[i]+=c;
}
int solve2(int l,int r,int c)
{
int s=0;int lg=belong[l],rg=belong[r];
if(lg==rg)
{
for(int i=l;i<=r;i++)
{
if(a[i]+add[lg]>=c)s++;
}
return s;
}
for(int i=l;i<=br[lg];i++)
{
if(a[i]+add[lg]>=c)s++;
}
for(int i=bl[rg];i<=r;i++)
{
if(a[i]+add[rg]>=c)s++;
}
for(int i=lg+1;i<=rg-1;i++)s+=ef(i,c-add[i]);
return s;
}
char s1[5];
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
fk();
for(int i=1;i<=belong
;i++)ps(i);
for(int i=1;i<=m;i++)
{
int x,y,c;
scanf("%s%d%d%d",s1+1,&x,&y,&c);
if(s1[1]=='M')solve1(x,y,c);
else printf("%d\n",solve2(x,y,c));
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  bzoj 分块 二分