leetcode 325. 和等于 k 的最长子数组长度 medium (重要)
题目描述:
给定一个数组 nums 和一个目标值 k,找到和等于 k 的最长子数组长度。如果不存在任意一个符合要求的子数组,则返回 0。
注意:
nums 数组的总和是一定在 32 位有符号整数范围之内的。
示例 1:
输入: nums = [1, -1, 5, -2, 3], k = 3
输出: 4
解释: 子数组 [1, -1, 5, -2] 和等于 3,且长度最长。
示例 2:
输入: nums = [-2, -1, 2, 1], k = 1
输出: 2
解释: 子数组 [-1, 2] 和等于 1,且长度最长。
进阶:
你能使时间复杂度在 O(n) 内完成此题吗?
解题思路:
通过这道题最大的收获就是理解前缀和数组的作用,利用前缀和我们能够快速计算某个子数组的和。
//利用前缀和数组 快速计算子数组的和
//而又因为求的是最长子数组等于k,所以我们在哈希表中保存的是前缀和为x第一次出现的下标
//然后我们求以每个元素为结尾的,最长 和等于k的子数组长度//值为x的前缀和最早出现的下标y hash【x】=y
unordered_map<int,int> hash;
// 这个初始化很重要,为了考虑到 0~i等于k的情况
hash[0]=-1;
代码:
//
class Solution {
public:
int maxSubArrayLen(vector<int>& nums, int k) {
//利用前缀和数组 快速计算子数组的和
//而又因为求的是最长子数组等于k,所以我们在哈希表中保存的是前缀和为x第一次出现的下标
//然后我们求以每个元素为结尾的,最长 和等于k的子数组长度
//值为x的前缀和最早出现的下标y hash【x】=y
unordered_map<int,int> hash;
// 这个初始化很重要,为了考虑到 0~i等于k的情况
hash[0]=-1;
int sum=0;
int res=0;
for(int i=0;i<nums.size();++i){
sum += nums[i];
if(hash.count(sum-k)) res= max(res,i-hash[sum-k]);
if(!hash.count(sum)) hash[sum]=i; // 因为要的是第一次出现的值
}
return res;
}
};