和为S（51Nod-2518）

题目

小b有一个01序列A，她想知道A有多少个非空连续子序列和为S。

你能帮帮她吗？

输入

第一行输入一个数n，表示A的长度；
第二行输入n个数‘0’或‘1’，表示A中的元素，以空格隔开；
第三行输入一个非负整数S；
其中0≤S≤n≤30000。

输出

输出一个数，表示子数组的个数

输入样例

5
1 0 1 0 1
2

输出样例

4

思路：维护一个前缀和即可

源程序

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#define EPS 1e-9
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
#define LL long long
const int MOD = 1E9+7;
const int N = 40000+5;
const int dx[] = {0,0,-1,1,-1,-1,1,1};
const int dy[] = {-1,1,0,0,-1,1,-1,1};
using namespace std;

int a[N];
int sum[N];
map<int,int> mp;
int main() {
    int n;
    scanf("%d",&n);
    for(int i=1; i<=n; i++){
        scanf("%d",&a[i]);
        sum[i]=a[i]+sum[i-1];
    }

    int k;
    scanf("%d",&k);
    int cnt=0;
    for(int i=1; i<=n; i++) {
        if(sum[i]==k)
            cnt++;

        int pos=sum[i]-k;
        cnt+=mp[pos];
        mp[sum[i]]++;
    }
    printf("%d\n",cnt);
    return 0;
}