Given an array of integers and an integer k,
you need to find the total number of continuous subarrays whose sum equals to k. Example 1: Input:nums = [1,1,1], k = 2 Output: 2 Note: The length of the array is in range [1, 20,000]. The range of numbers in the array is [-1000, 1000] and the range of the integer k is [-1e7, 1e7].
subarray sum 问题常用hashmap, 存count 值和坐标, 动归的感觉啊
public int subarraySum(int[] nums, int k) { int sum = 0, result = 0; Map<Integer, Integer> preSum = new HashMap<>(); preSum.put(0, 1); for (int i = 0; i < nums.length; i++) { sum += nums[i]; if (preSum.containsKey(sum - k)) { result += preSum.get(sum - k); } // 当加着加着出现两个一样的sum时, 要在他的value上加1, 因为可以有多个连续的串 preSum.put(sum, preSum.getOrDefault(sum, 0) + 1); } return result; }
要用 preSum.put(0, 1); 是得result 加的值可以来自map中的多个.
不能 if (sum == k) {
result++;
}
因为:
public boolean checkSubarraySum(int[] nums, int k) { Map<Integer, Integer> map = new HashMap<Integer, Integer>(){{put(0,-1);}};; int runningSum = 0; for (int i=0;i<nums.length;i++) { runningSum += nums[i]; if (k != 0) runningSum %= k; Integer prev = map.get(runningSum); if (prev != null) { if (i - prev > 1) return true; } else map.put(runningSum, i); } return false; }
原文:http://www.cnblogs.com/apanda009/p/7282555.html