2022-5-18[Hash] 非空连续子序列个数

CiLin-Yan

已于 2022-05-21 08:32:51 修改

阅读量251

点赞数

分类专栏：刷题记录文章标签： c++ 算法

于 2022-05-18 10:18:43 首次发布

本文链接：https://blog.csdn.net/weixin_43791477/article/details/124836923

版权

刷题记录专栏收录该内容

70 篇文章 0 订阅

订阅专栏

这篇博客介绍了一个关于01序列的算法问题，即计算所有非空连续子序列和等于给定值S的子序列数量。通过使用动态规划的方法，程序可以有效地找出所有满足条件的子序列，其核心代码展示了如何利用unordered_map存储前缀和并计算子序列个数。这是一个典型的动态规划应用实例，适合对算法和数据结构感兴趣的读者深入理解。

摘要由CSDN通过智能技术生成

题目

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

#define N 30005
#pragma GCC optimize(2)
using namespace std;
int val[N];

int main() {
    ios::sync_with_stdio(false);
    int n, target, ans = 0;
    cin >> n;
    for (int i = 0; i < n; ++i) { cin >> val[i]; }
    cin >> target;
    unordered_map<int, int> umap;
    umap[0] = 1;
    for (int i = 0, curSum = 0; i < n; ++i) {
        curSum += val[i];
        ans += umap[curSum - target];
        umap[curSum]++;
    }
    cout << ans;
    return 0;
}