未排序数组中累加和为给定值的最长子数组长度

未排序数组中累加和为给定值的最长子数组长度

题目描述：

给定一个无序数组arr, 其中元素可正、可负、可0。给定一个整数k，求arr所有子数组中累加和为k的最长子数组长度

输入描述：

第一行两个整数N, k。N表示数组长度，k的定义已在题目描述中给出
第二行N个整数表示数组内的数

输出描述：

输出一个整数表示答案

示例1

输入

5 0
1 -2 1 1 1

输出

3

备注：

$1\le N\le 10^5$

$-10^9\le k\le 10^9$

$-100\le a_i\le 100$

---

题解：

核心是前缀和。设 s[i] 表示 arr[0…i] 的和，则 s[j+1…i] = s[i] - s[j]。假设我们当前遍历到位置 i ，如果我们能找到前面的和为 s[i]-k 的子数组右边界，则 arr[j+1…i] 的和即为 k ，也就是找到一组解。这意味着我们需要一个哈希表保存前缀和中每个元素出现的位置，求最大值的话，则需要保存最早出现的位置即可。

注意：需要添加一个 (0,-1) 到哈希表中，避免漏了从 0 开始的子数组。

代码：

#include <cstdio>
#include <unordered_map>

using namespace std;

int n, k;
int val;
unordered_map<int, int> _hash;

int main(void) {
    scanf("%d%d", &n, &k);
    int sum = 0;
    _hash[0] = -1;
    int ret = 0, t;
    for ( int i = 0; i < n; ++i ) {
        scanf("%d", &val);
        sum += val;
        t = sum - k;
        if ( _hash.count(t) ) {
            ret = max(ret, i - _hash[t]);
        }
        if (!_hash.count(sum)) _hash[sum] = i;
    }
    return 0 * printf("%d\n", ret);
}