题意:给你一堆数字,每个数字有正负之分,求任意区间内和为k的子区间的个数。
题解:
先把前缀和都求出来,构成一个数组sum。
建立一个hash表,然后考虑区间sum[l,r],从左到右扫,每扫到一个前缀sum[i],从把答案加上hash_table[sum[i] - k],并且hash_table[sum[i-1]]++;
这是对于固定区间的做法,要用莫队得证明区间左右端点移动的时间复杂度为O(1):
当l增大1时,ans -= hash_table[sum[l-1]+k],(即把sum[l-1]的贡献消除)hash_table[sum[l-1]]--,l++;
当l减小1时,l--,ans += hash_table[sum[l-1]+k](即把sum[l-1]的贡献加上)hash_table[sym[l-1]]++;
。。。。
然后就ojbk了
代码:
#include <bits/stdc++.h>
#include<tr1/unordered_map>
using namespace std;
#define int long long
int n,k,q,sn;
const int maxn = 100007;
int a[maxn];
int sum[maxn],res[maxn];
int qcnt = 0;struct work{int id,l,r;bool operator<(const work &q)const{int bk1 = (l-1)/sn,bk2 = (q.l-1)/sn;if(bk1 == bk2)return r < q.r;return bk1 < bk2;}
}qs[maxn];
tr1::unordered_map<int,int>mp;
main(){ scanf("%lld%lld",&n,&k);sn = 2*(int)sqrt(n+0.5);for(int i = 1;i <= n;++i) {int tmp;scanf("%lld",&tmp);a[i] = tmp == 1?1:-1;}for(int i = 1;i <= n;++i){int tp;scanf("%lld",&tp);a[i] *= tp;sum[i] = sum[i-1] + a[i];}scanf("%lld",&q);while(q--){int l,r;scanf("%lld%lld",&l,&r);qs[qcnt] = (work){qcnt,l,r};qcnt++;}sort(qs,qs+qcnt);int l = -1,r = -1;int ans = 0;//mp[0] = 1;for(int i = 0;i < qcnt;++i){if(l == -1) l = qs[i].l,r = qs[i].l-1,mp[sum[qs[i].l-1]] ++;for(;l < qs[i].l;l++){mp[sum[l-1]] --;ans -= mp[sum[l-1]+k];}for(r++;r <= qs[i].r;r++){ans += mp[sum[r]-k];mp[sum[r]]++;}--r;for(--l;l >= qs[i].l;--l){ans += mp[sum[l-1]+k];mp[sum[l-1]]++;}++l;for(r;r > qs[i].r;--r){ mp[sum[r]]--;ans -= mp[sum[r]-k];}res[qs[i].id] = ans;}for(int i = 0;i < qcnt;i++){printf("%lld\n",res[i]);}return 0;
}