【Leetcode】325. Maximum Size Subarray Sum Equals k

题目地址：

https://leetcode.com/problems/maximum-size-subarray-sum-equals-k/

给定一个数组，求其最长的和为$k$的子区间的长度。

由于是求子区间的问题，容易想到用前缀和数组来使得查询效率更高。暴力的做法无非就是枚举左端点，然后移动右端点，同时维护一个变量保存当前得到的和为$k$的最长区间的长度。为了提高效率，可以用一个哈希表来保存前缀和及其下标（前缀和数组$a[i]$一般保存的是前$i$个数字的和，并且$i$从$0$开始。这道题里可以令空前缀的右端点下标是$-1$），这样每次查询的时候只需要check当前的前缀和减去$k$是否在哈希表中就可以了。代码如下：

class Solution {
 public:
  using ll = long long;
  int maxSubArrayLen(vector<int>& a, int k) {
    int res = 0;
    unordered_map<int, int> mp;
    mp[0] = -1;
    ll s = 0;
    for (int i = 0; i < a.size(); i++) {
      s += a[i];
      if (mp.count(s - k)) res = max(res, i - mp[s - k]);
      if (!mp.count(s)) mp[s] = i;
    }
    return res;
  }
};

时空复杂度$O(n)$。