和为S

2518 和为S

1. 2 秒
 
2. 262,144 KB
 
3. 10 分
 
4. 2 级题

 

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

你能帮帮她吗？

 收起

 

输入

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

输出

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

输入样例

5
1 0 1 0 1
2

输出样例

4
思路：f[i]为前i相和，然后暴力做一些n方复杂度可解

#include<iostream>
#include<algorithm>
using namespace std;
int a,n,k,f[30005],sum;
int main(){
cin>>n;
for(int i=1;i<=n;i++){
    cin>>a;
    f[i]=f[i-1]+a;
}
cin>>k;
for(int i=0;i<=n-k;i++){
    for(int j=i+k;j<=n;j++){
        if(f[j]-f[i]==k&&i!=j)
        sum++;
    
    }
}
cout<<sum<<endl;
    return 0;
}