传送门
题解就不说了。
感觉官方题解3很妙,由此可以归纳出一个通用方法,以解决支持区间修改,单点查询历史中满足某个条件的个数这一类问题。
据说有黑魔法的
(n+q)n√
的做法,但不如算法三+将块设为
n√
快?
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=1000010,M=3000;
typedef long long ll;
int n,q,a[N],i,l,r,x,p,y,o,xb,ans[N];
struct query{
int o,i,t,x;
bool operator<(const query&x)const{
return i==x.i?o<x.o:i<x.i;
}
}b[N];
struct block{
int sz,be[M],en[M],i,x,y,z,num[N],bs,j,k,ans;
pair<ll,int> a[N],b[N],c[N];
ll tag[M];
inline void init(){
sz=sqrt(q);
for(i=1;i<=q;++i)num[i]=(i-1)/sz+1,a[i]=make_pair(0,i);
bs=num[q];
be[1]=1;
en[1]=sz;
for(i=2;i<bs;++i)be[i]=be[i-1]+sz,en[i]=en[i-1]+sz;
be[bs]=en[bs-1]+1;
en[bs]=q;
}
inline void add(int p,ll v){
x=num[p];
for(i=x+1;i<=bs;++i)tag[i]+=v;
for(i=be[x],y=z=0;i<=en[x];++i)
if(a[i].second>=p)b[++y]=make_pair(a[i].first+v,a[i].second);
else c[++z]=a[i];
for(i=j=1,k=be[x];i<=y && j<=z;)a[k++]=b[i]<c[j]?b[i++]:c[j++];
for(;i<=y;)a[k++]=b[i++];
for(;j<=z;)a[k++]=c[j++];
}
inline int query(int p,int v){
if(!p)return 0;
x=num[p];
for(i=1,ans=0;i<x;++i)
ans+=sz+1-(lower_bound(a+be[i],a+en[i]+1,make_pair(1ll*v-tag[i],0))-a-be[i]+1);
for(i=be[x];i<=en[x];++i)ans+=a[i].second<=p && 1ll*a[i].first+tag[x]>=v;
return ans;
}
}B;
int main(){
scanf("%d%d",&n,&q);
for(i=1;i<=n;++i)scanf("%d",a+i);
for(i=1;i<=q;++i){
scanf("%d",&o);
if(o&1){
scanf("%d%d%d",&l,&r,&x);
b[++xb]=(query){o,l,i,x};
b[++xb]=(query){o,r+1,i,-x};
}else{
scanf("%d%d",&p,&y);
b[++xb]=(query){o,p,i,y};
}
}
sort(b+1,b+xb+1);
B.init();
for(i=1;i<=q;++i)ans[i]=-1;
for(i=1;i<=xb;++i)
if(b[i].o&1)B.add(b[i].t,b[i].x);
else ans[b[i].t]=B.query(b[i].t-1,1ll*b[i].x-a[b[i].i])+(a[b[i].i]>=b[i].x);
for(i=1;i<=q;++i)if(ans[i]>=0)printf("%d\n",ans[i]);
return 0;
}