#6278. 数列分块入门 2(询问区间内小于某个值 xx 的元素个数)

时间:2025-01-04 22:35:14

题目链接:https://loj.ac/problem/6278

题目大意:中文题目

具体思路:数列分块模板题,对于更新的时候,我们通过一个辅助数组来进行,对于原始的数组,我们只是用来加减,然后这个辅助数组的作用就是对每一块进行排序,当查询的时候,如果不是整块的,我们直接通过a数组来记录,对于整块的,我们直接通过排序的辅助数组进行二分查找就可以了。

注意每一次更新,除了整块的,我们都需要进行对b数组这一整块的重新赋值,因为只是部分赋值的话,b数组已经排序了,这样的话数组的下标对应的数就已经改变了。

lower_bound查询的时候, id=lower_bound(b+l[i],b+r[i]+1,tmp)-(b+l[i]);如果是需要记录小于的个数的话,直接减去b+l[i]就是个数了。

A代码:

 #include<bits/stdc++.h>
using namespace std;
# define ll long long
const int maxn = 2e5+;
const int mod = 1e8+;
ll l[maxn],r[maxn],belong[maxn];
ll add[maxn],a[maxn],b[maxn];
int n;
void buildblock()
{
ll tmp=(ll)sqrt(n);
ll tot=n/tmp;
if(n%tmp)
tot++;
for(ll i=; i<=n; i++)
{
belong[i]=(i-)/tmp+1ll;
}
for(ll i=; i<=tot; i++)
{
l[i]=(i-)*tmp+1ll;
r[i]=i*tmp;
}
r[tot]=n;
for(ll i=; i<=tot; i++)
{
sort(b+l[i],b+r[i]+);
}
}
void init(int tmp){
for(int i=l[tmp];i<=r[tmp];i++)b[i]=a[i];
sort(b+l[tmp],b+r[tmp]+);
}
void update(ll st,ll ed,ll val)
{
if(belong[st]==belong[ed])
{
for(ll i=st; i<=ed; i++)a[i]+=val;
init(belong[st]);
return ;
}
for(ll i=st; i<=r[belong[st]]; i++)a[i]+=val;
init(belong[st]);
for(ll i=l[belong[ed]]; i<=ed; i++)a[i]+=val;
init(belong[ed]);
for(ll i=belong[st]+; i<belong[ed]; i++)
add[i]+=val;
}
ll ask(ll st,ll ed,ll val)
{
ll ans=;
if(belong[st]==belong[ed])
{
for(ll i=st; i<=ed; i++)
{
if(a[i]+add[belong[st]]<val)
ans++;
}
return ans;
}
for(ll i=st; i<=r[belong[st]]; i++)
{
ans+=(a[i]+add[belong[st]]<val?:);
}
for(ll i=l[belong[ed]]; i<=ed; i++)
{
ans+=(a[i]+add[belong[ed]]<val?:);
}
for(ll i=belong[st]+; i<belong[ed]; i++)
{
ll tmp=val-add[i];
ll id=lower_bound(b+l[i],b+r[i]+,tmp)-(b+l[i]);
ans+=id;
}
return ans;
}
int main()
{
// freopen("hqx.in","r",stdin);
scanf("%d",&n);
for(int i=; i<=n; i++)
{
scanf("%lld",&a[i]);
b[i]=a[i];
}
buildblock();
ll op,st,ed;
ll val;
while(n--)
{
scanf("%lld %lld %lld %lld",&op,&st,&ed,&val);
if(op==)
{
update(st,ed,val);
}
else if(op==)
{
printf("%lld\n",ask(st,ed,val*val));
}
}
return ;
}