正题
题目链接:https://www.luogu.com.cn/problem/P2801
题目大意
nnn个数字,要求支持
- 区间加上一个数字www
- 询问一个区间内不小于www的数的个数
解题思路
考虑分块,对于块内我们维护一个排序后的数组,查询时直接在整个块中二分答案即可。修改就暴力修改然后排序。
时间复杂度O(Qnlogn)O(Q\sqrt n\log n)O(Qnlogn)
codecodecode
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=1e6+10;
int n,m,T,a[N],v[N],pos[N];
int L[N],R[N],lazy[N];
void Reset(int x){for(int i=L[x];i<=R[x];i++)a[i]+=lazy[x],v[i]=a[i];sort(v+L[x],v+R[x]+1);lazy[x]=0;return;
}
void seq_qer(){for(int i=1;i<=T;i++)L[i]=R[i-1]+1,R[i]=i*T;if(R[T]!=n)L[++T]=R[T]+1,R[T]=n;for(int i=1;i<=T;i++)for(int j=L[i];j<=R[i];j++)pos[j]=i;for(int i=1;i<=T;i++)Reset(i);
}
int Ask(int l,int r,int w){int q=pos[l],p=pos[r],ans=0;if(q==p){for(int i=l;i<=r;i++)ans+=(a[i]+lazy[p]>=w);return ans;}for(int i=l;i<=R[q];i++)ans+=(a[i]+lazy[q]>=w);for(int i=L[p];i<=r;i++)ans+=(a[i]+lazy[p]>=w);for(int i=q+1;i<p;i++){if(v[R[i]]+lazy[i]<w)ans+=R[i]-L[i]+1;else ans+=R[i]+1-(lower_bound(v+L[i],v+R[i]+1,w-lazy[i])-v);}return ans;
}
void Change(int l,int r,int w){int q=pos[l],p=pos[r],ans=0;if(q==p){for(int i=l;i<=r;i++)a[i]+=w;Reset(p);return;}for(int i=l;i<=R[q];i++)a[i]+=w;Reset(q);for(int i=L[p];i<=r;i++)a[i]+=w;Reset(p);for(int i=q+1;i<p;i++)lazy[i]+=w;return;
}
int main()
{scanf("%d%d",&n,&m);for(int i=1;i<=n;i++)scanf("%d",&a[i]);T=sqrt(n);seq_qer();while(m--){printf("%d ",n);for(int i=1;i<=n;i++)printf("%d ",v[i]);char op[3];int l,r,w;scanf("%s %d %d %d",op,&l,&r,&w);if(op[0]=='M')Change(l,r,w);else printf("%d\n",Ask(l,r,w));}return 0;
}