题意:
解法:
容易想到将t[i]=2的书的值a[i]变为-a[i],
那么合法的区间只需要满足区间和=k.
考虑一个问题:
整个序列有多少个满足条件的子区间.
预处理前缀和,枚举右端点r,合法的左端点满足sum[r]-sum[l-1]=k,
用桶记录一下每种sum[l-1]的个数即可.
将询问(l,r)变为(l-1,r),那么只需要查询sum[r]-sum[l]=k,这样写比较方便处理.
这题可以离线,而且还是区间查询,因此可以考虑莫队,
当移动左端点l时,需要处理sum[l]+k的个数,
当移动右端点r时,需要处理sum[r]-k的个数.
明白如何处理端点移动时对答案的贡献后,剩下的部分就是简单莫队了.
注意:
由于sum[i]比较大,所以需要用map存,这样会多一个log的复杂度,然后就超时了,
更优秀的做法是先离散化一下,这样就不会多出这一个log了.
code:
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=4e5+5;
const int block=1000;
struct QQ{
int l,r,id;
}Q[maxm];
int mp[maxm];
int res[maxm];
int sum[maxm];
int t[maxm];
int a[maxm];
int n,k;
int ans;
bool cmp(QQ a,QQ b){
if(a.l/block!=b.l/block)return a.l<b.l;
return a.r<b.r;
}
//离散化
int xx[maxm];
int val[maxm];
int L[maxm],R[maxm];
void init(){
int num=0;
for(int i=0;i<=n;i++){
xx[++num]=sum[i];
xx[++num]=sum[i]+k;
xx[++num]=sum[i]-k;
}
sort(xx+1,xx+1+num);
num=unique(xx+1,xx+1+num)-xx-1;
for(int i=0;i<=n;i++){
val[i]=lower_bound(xx+1,xx+1+num,sum[i])-xx;
L[i]=lower_bound(xx+1,xx+1+num,sum[i]+k)-xx;
R[i]=lower_bound(xx+1,xx+1+num,sum[i]-k)-xx;
}
}
void solve(){
cin>>n>>k;
for(int i=1;i<=n;i++)cin>>t[i];
for(int i=1;i<=n;i++){
cin>>a[i];
if(t[i]==2)a[i]=-a[i];
}
for(int i=1;i<=n;i++){
sum[i]=sum[i-1]+a[i];
}
init();
int q;cin>>q;
for(int i=1;i<=q;i++){
int l,r;cin>>l>>r;
Q[i]={l-1,r,i};
}
sort(Q+1,Q+1+q,cmp);
int l=Q[1].l,r=l-1;
for(int i=1;i<=q;i++){
int ql=Q[i].l,qr=Q[i].r;
while(r>qr){
mp[val[r]]--;
ans-=mp[R[r]];
r--;
}
while(l<ql){
mp[val[l]]--;
ans-=mp[L[l]];
l++;
}
while(r<qr){
r++;
ans+=mp[R[r]];
mp[val[r]]++;
}
while(l>ql){
l--;
ans+=mp[L[l]];
mp[val[l]]++;
}
res[Q[i].id]=ans;
}
for(int i=1;i<=q;i++){
cout<<res[i]<<endl;
}
}
signed main(){
ios::sync_with_stdio(0);cin.tie(0);
solve();
return 0;
}