Leetcode 560 Subarray Sum Equals K
Using Cummulative sum
sum[i] is used to store the cumulative sum of numsnums array upto the element corresponding to the (i-1)th.
th
index.
for (int i = 1; i <= nums.length; i++)
sum[i] = sum[i - 1] + nums[i - 1];
for (int start = 0; start < nums.length; start++) {
for (int end = start + 1; end <= nums.length; end++) {
if (sum[end] - sum[start] == k)
count++;
}
}
Without space
for (int start = 0; start < nums.length; start++) {
int sum=0;
for (int end = start; end < nums.length; end++) {
sum+=nums[end];
if (sum == k)
count++;
}
}
Using hashmap
HashMap < Integer, Integer > map = new HashMap < > ();
map.put(0, 1);
for (int i = 0; i < nums.length; i++) {
sum += nums[i];
if (map.containsKey(sum - k))
count += map.get(sum - k);
map.put(sum, map.getOrDefault(sum, 0) + 1);
}
For Example:
Input:nums = [1,1,1], k = 2
Output: 2
- Map: (0,1)
- Map: (0,1), (1,1)
- Map: (0,1), (1,1), (2,1) count =1
- Map: (0,1), (1,1), (2,1), (3,1) count =2