小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<bits/stdc++.h>
using namespace std;
int a[30010];
int main()
{
int n,k,sum=0;
cin>>n;
a[0]=0;
for(int i=1;i<=n;i++)
{
cin>>a[i];
a[i]+=a[i-1];
}
cin>>k;
for(int i=1;i<=n;i++)
{
if(a[i]==k) sum++;
for(int j=1;j<i;j++)
{
if(a[i]-a[j]<k) break;
if(a[i]-a[j]==k) sum++;
}
}
cout<<sum;
return 0;
}