简单题不简单之和为S

小 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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值