題意：給你一堆數(shù)字，每個數(shù)字有正負之分，求任意區(qū)間內(nèi)和為k的子區(qū)間的個數(shù)。

題解：

先把前綴和都求出來，構(gòu)成一個數(shù)組sum。

建立一個hash表，然后考慮區(qū)間sum[l,r]，從左到右掃，每掃到一個前綴sum[i]，從把答案加上hash_table[sum[i] - k]，并且hash_table[sum[i-1]]++;

這是對于固定區(qū)間的做法，要用莫隊得證明區(qū)間左右端點移動的時間復雜度為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; }

總結(jié)

以上是生活随笔為你收集整理的codeforces 877F F. Ann and Books hash+莫队算法的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。