小 b 有一个 01 序列 A ,她想知道 A 有多少个非空连续子序列和为 S 。
你能帮帮她吗?
输入:
第一行输入一个数n,表示A的长度;
第二行输入 nn 个数 ′0′ 或 ′1′,表示A 中的元素,以空格隔开;
第三行输入一个非负整数 S ;
其中 0≤S≤n≤300000 。
输出:
输出一个数,表示子数组的个数。
样例输入:
5 1 0 1 0 1 2
样例输出:
4
不多废话,写好初始代码:
#include<bits/stdc++.h>
using namespace std;
int n,s[30010],m,ans;//我把数组A定义为s,把整数S定义为m
int main(){
return 0;
}
输入比较特殊,而且把下一步一起做了,但也挺简单:
cin>>n;//输入n
for(int i=1;i<=n;i++){
int x;
cin>>x;
s[i]=s[i-1]+x;//这就是下一步要做的,输入之内完成了
}
cin>>m;//输入原先的整数S
代码主干部分,可以用双重循环:
for(int i=1;i<=n;i++){
for(int j=i;j<=n;j++){
if(s[j]-s[i-1]==m) ans++;//累加子数组个数
if(s[j]-s[i-1]>m) break;//加完了退出循环,之后输出ans
}
}
输出就一行:
cout<<ans;
又是喜闻乐见的“缝合”环节,我把标语去掉了:
#include<bits/stdc++.h>
using namespace std;
int n,s[30010],m,ans;
int main(){
cin>>n;
for(int i=1;i<=n;i++){
int x;
cin>>x;
s[i]=s[i-1]+x;
}
cin>>m;
for(int i=1;i<=n;i++){
for(int j=i;j<=n;j++){
if(s[j]-s[i-1]==m) ans++;
if(s[j]-s[i-1]>m) break;
}
}
cout<<ans;
return 0;
}