51nod 2518 和为S

2518 和为S

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

你能帮帮她吗？

 收起

输入

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

输出

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

输入样例

5
1 0 1 0 1
2

输出样例

4

Code:

#include <bits/stdc++.h>
using namespace std;
int a[30008];
int main() { 
    int n,s,sum=0,ans=0;
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    cin>>s;
    for(int i=0;i<n;i++){
        sum=a[i];
        for(int j=i+1;j<=n;j++){
            if(sum==s)    ans++;
            sum+=a[j];
        }
    }
    cout<<ans;
}