Codeforces877 F. Ann and Books（莫队+离散化）

题意：

解法：

容易想到将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;
}

---