This straightforward approach involves checking all possible subarrays in the given array. We calculate the sum for each subarray and check if it equals k. Although simple, it can be inefficient for large arrays due to its O(n^2) time complexity.
Time Complexity: O(n^2)
Space Complexity: O(1)
1#include <stdio.h>
2
3int subarraySum(int* nums, int numsSize, int k) {
4 int count = 0;
5 for (int start = 0; start < numsSize; start++) {
6 int sum = 0;
7 for (int end = start; end < numsSize; end++) {
8 sum += nums[end];
9 if (sum == k) count++;
10 }
11 }
12 return count;
13}
14
15int main() {
16 int nums[] = {1, 1, 1};
17 int k = 2;
18 int size = sizeof(nums)/sizeof(nums[0]);
19 printf("%d\n", subarraySum(nums, size, k));
20 return 0;
21}
22This C code implements a nested loop where the outer loop selects the start of the subarray and the inner loop expands it, calculating the sum. If the sum equals k, it increments the count. The time complexity is O(n^2), and the space complexity is O(1).
This optimized approach uses a hashmap to store prefix sums, facilitating the identification of subarrays with the desired sum in constant time. By keeping a count of prefix sums that have been seen, we can determine how many times a specific sum - k has appeared, suggesting that a subarray ending at the current position sums to k.
Time Complexity: O(n)
Space Complexity: O(n)
1from collections import defaultdict
2
3def subarray_sum(nums, k):
4 prefix_sum_count = defaultdict(int)
5 prefix_sum_count[0] = 1
6 count = sum_ = 0
7
8 for num in nums:
9 sum_ += num
10 count += prefix_sum_count[sum_ - k]
11 prefix_sum_count[sum_] += 1
12
13 return count
14
15# Example usage
16nums = [1, 1, 1]
17k = 2
18print(subarray_sum(nums, k))
19This Python solution utilizes a defaultdict to store prefix sums counts, allowing for dynamic and efficient querying of existing sums and direct summation of subarray occurrences satisfying the sum condition. The time and space complexities are O(n).