leetcode学习打卡--560. 和为K的子数组(蛮力搜索/哈希表优化)

560. 和为K的子数组

给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

---

前台样例

示例 1 :
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。


说明 :
数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。

---

题目分析:
今天的题还是需要思考的
1.题目说明了要求子数组连续 我们不能断开它。
2.示例1说明了 情况的不同取决于子数组段位置 而不是子数组本身 这样其实更方便求解。
3.数组极限长度 20000 O(n^2)的代码可能超时。

---

解体思路:
1.当没有思路的时候首先采用蛮力法思考

初始思路–对于数组的每一个元素均向后遍历(以该元素为头)，若满足则break，计数器加一，若大于目标了直接break。

//WA版蛮力法
class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int len=nums.size();
        int res=0;
        for(int i=0;i<len;i++)
        {
            int sum=0;
            for(int j=i;j<len;j++)
            {
                sum+=nums[j];
                if(sum==k) {res++;break;}
                if(sum>k) break;
            }
        }
        return res;
    }
};

提交 答案错误

34 / 81 个通过测试用例

输入：
[28,54,7,-70,22,65,-6]
100
输出：
0
预期：
1

看到负数就明白 有可能累加和大于了目标值但还可以减下了，所以不对。
同理 如果等于了 也不能break，因为后面还有可能出现满足的。

所以 AC蛮力法如下 时间1612ms居然过了

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int len=nums.size();
        int res=0;
        for(int i=0;i<len;i++)
        {
            int sum=0;
            for(int j=i;j<len;j++)
            {
                sum+=nums[j];
                if(sum==k) res++;
            }
        }
        return res;
    }
};

---

那么蛮力法的缺点其实很明显了，重复计算的段太多了。
优化应该采用hash表查找
copy官方题解了

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int, int> mp;
        mp[0] = 1;
        int count = 0, pre = 0;
        for (auto& x:nums) {
            pre += x;
            if (mp.find(pre - k) != mp.end()) count += mp[pre - k];
            mp[pre]++;
        }
        return count;
    }
};

用pre来存储序列累加合，譬如pre[i]代表的就是从第0个数到第i个数的累加和
而map其实对应着序列段长度以及其出现的次数。
举个例子

数组为: 3,4,-4,4,7,2,-3,1
k值为: 7

当我们逐位遍历直到1时 此时 pre=14 14-k=7
而前面出现7的地方有 3+4=7 3+4-4+4=7 所以此时应该加2.也就是count += mp[pre - k];

也就是说 如果一个数段能分为3段 a-b-c
我们要让k=7 也就是bc=7 而ac=14 所以ab=14-7=7 只要找到了所以满足的ab段也就对应着所有解了。