CF877F题解

题目大意

有一个序列，每个位置上有 \(1\) 或 \(2\) 两种元素若干，每次询问一个区间，求这个区间有多少个子区间满足 \(1\) 类元素恰好比 \(2\) 类元素多 \(k\) 个。

莫队

要是看到很难维护的东西，又允许离线，多半就是莫队。（雾）

来看看一个区间在满足条件的情况有啥可以用到的东西：

\[s1[R] - s1[L-1] = s2[R] - S2[L-1] + k
\]

\(s1[i]\) 表示从 \(1\) 到 \(i\) 有多少个一类元素，\(s2\) 同理。

移项：

\[(s1[R]-s2[R])=(s1[L-1]-s2[L-1])+k
\]

转化成了这样一个问题：

\[\sum_{i=L-1}^{R-1}\sum_{j=i+1}^R[val[j]=val[i]+k]
\]

其中 \(val[i]\) 表示 \(s1[i] - s2[i]\)。

首先我们在莫队的过程中维护一个桶，表示目前区间中有多少个 \(val[i]\)。

在莫队的转移过程中，我们得考虑新加入的元素是在左侧还是在右侧。

如果在左侧，\(ans+=Bucket[val[i]+k]\)（因为区间的右侧要比自己大）；如果在右侧，则 \(ans+=Bucket[val[i]-k]\)。

关于如何维护桶，离散化就行了。

记得离散化的数组不仅要加入 \(val[i],val[i]+k\) 和 \(val[i]-k\)，还要加入 \(0,k\) 和 \(-k\)。（因为 \(val[0]=0\))

code:

#include<algorithm>
#include<cstdio>
#include<cmath>
const int M=1e5+5;
typedef long long ll;
int len;ll lsh[M*3];
int n,m,k,p,t[M],v[M],nxt[M],pre[M];ll CB[M*3];ll now,val[M],ans[M];
struct Query{
	int L,R,p,id;
	inline bool operator<(const Query&it)const{
		return p==it.p?p&1?R<it.R:R>it.R:L<it.L;
	}
}q[M];
inline void Add(const int&id,const int&to){
	now+=CB[to];++CB[v[id]];
}
inline void Del(const int&id,const int&to){
	--CB[v[id]];now-=CB[to];
}
signed main(){
	register int i,L=1,R=0;
	scanf("%d%d",&n,&k);lsh[++len]=0;lsh[++len]=k;lsh[++len]=-k;
	for(i=1;i<=n;++i)scanf("%d",t+i);
	for(i=1;i<=n;++i){
		scanf("%lld",val+i);
		if(t[i]==1)val[i]=val[i-1]+val[i];
		else val[i]=val[i-1]-val[i];
		lsh[++len]=val[i];lsh[++len]=val[i]-k;lsh[++len]=val[i]+k;
	}
	std::sort(lsh+1,lsh+len+1);len=std::unique(lsh+1,lsh+len+1)-lsh-1;
	for(i=0;i<=n;++i){
		v[i]=std::lower_bound(lsh+1,lsh+len+1,val[i])-lsh;
		pre[i]=std::lower_bound(lsh+1,lsh+len+1,val[i]-k)-lsh;
		nxt[i]=std::lower_bound(lsh+1,lsh+len+1,val[i]+k)-lsh;
	}
	scanf("%d",&m);p=ceil(n/sqrt(m*2.0/3));
	for(i=1;i<=m;++i){
		scanf("%d%d",&q[i].L,&q[i].R);--q[i].L;
		q[i].id=i;q[i].p=q[i].L/p;
	}
	std::sort(q+1,q+m+1);
	for(i=1;i<=m;++i){
		const int&QL=q[i].L,&QR=q[i].R;
		while(R<QR)++R,Add(R,pre[R]);
		while(L>QL)--L,Add(L,nxt[L]);
		while(L<QL)Del(L,nxt[L]),++L;
		while(R>QR)Del(R,pre[R]),--R;
		ans[q[i].id]=now;
	}
	for(i=1;i<=m;++i)printf("%lld\n",ans[i]);
}