Leetcode之Subarray Sum Equals K问题

来源:互联网 发布:小米机械键盘 mac 编辑:程序博客网 时间:2024/05/16 17:58

问题描述:

Given an array of integers and an integer k, you need to find the total number ofcontinuous subarrays whose sum equals to k.

Note:

  1. The length of the array is in range [1, 20,000].
  2. The range of numbers in the array is [-1000, 1000] and the range of the integer k is [-1e7, 1e7].

示例一:

Input:nums = [1,1,1], k = 2
Output: 2

问题来源:Subarray Sum Equals K (详细地址:https://leetcode.com/problems/subarray-sum-equals-k/description/)

思路分析:在这介绍一种用Hashmap来解决战斗的方案,其中的key我们放的是从0到当前索引所有元素的和sum,value是对应的sum所出现的次数,为什么我们这么做呢?其中的道理是啥呢?假设当前遍历到的数组下标为j,则:

sum[0...j]-sum[0...i-1]=sum[i...j];

看懂了吗?要找到元素之和等于k的子数组,显然在这对应的项是等式右边的那一项(sum[i....j]),我们把第一项看成sum,那么得到第二项(sum[0...i-1])为sum-k,所以我们只需要判断整个的hashmap中有没有存放sum-k的值就可以,存在的话我们就取出它对应的次数加上即可,接着把当前遍历到j的和存放进去;当然,没有出现的话那我们也得把当前的和添加到map中存好。最后,我们介绍一下初始化的问题,也就是map中和为0的初始值应该是多少?是0吗?不急,我们先看看map中的和为0是啥意思?也就是说sum=k,也就是遍历到当前索引j的时候正好和等于k,所以初值不应该是0而是1,最后得累加到最终的结果result中去。

代码: