【数据结构与算法】前缀和+哈希表算法

一、引入

关于前缀和和哈希这两个概念大家都不陌生，在之前的文章中也有过介绍：前缀和与差分算法详解

而哈希表最经典的一题莫过于两数之和
题目链接

题目描述：

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target 的那 两个 整数，并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

只会存在一个有效答案

思路分析：
我们在遍历这个数组要做两件事：
假设现在遍历到下标为idx的位置。
1️⃣ 查看target - nums[idx]是否在哈希表中，如果在，说明这两个数加起来就是目标和，那么就找到了两个下标，一个是hash[target - nums[idx]]，一个是当前位置idx。
2️⃣ 用哈希表记录两个数据，first记录当前位置的值，second记录当前位置的下标。

代码：

class Solution {
   
public:
    vector<int> twoSum(vector<int>& nums, int target) {
   
        unordered_map<int, int> hash;
        int n = nums.size();
        for(int i = 0; i < n; i++)
        {
   
            if(hash.find(target - nums[i]) != hash.end())
            {
   
                return {
   hash[target - nums[i]], i};
            }
            hash[nums[i]] = i;
        }
        return {
   };
    }
};

二、前缀和与哈希表的结合

用一个例子来说明以下：

假设我们要寻和为5的连续子数组的个数，那么只要前缀和中任意两个数的差值为5，那么就找到了子数组。

那么我们就可以直接用哈希表把前缀和的数据存储起来，first存前缀和的值，second用来标识有多少个子数组。
这里首先要注意初始化哈希表把0的位置先设置成1：hash[0] = 1，因为当我们计算前缀和为5的位置的时候，就标识了从0 ~ 5存在和为5的连续子数组。

假设目标和为k，遍历到i位置。
所以现在我们在计算前缀和的同时看看是否存在hash[k - nums[i]]，这个的数值大小就代表有多少个连续的子数组和。那么为什么会存在多个呢？

因为可能数组存在负数，这样就会导致出现这种情况：

那么省略号这段区间的前缀总和就为0，所以就会存在两段子数组和为5的区间。