Create 0560-subarray-sum-equals-k.kt

nick322 · Jan 28, 2023 · c1eb794 · c1eb794
1 parent 669079d
commit c1eb794
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/kotlin/0560-subarray-sum-equals-k.kt b/kotlin/0560-subarray-sum-equals-k.kt
@@ -0,0 +1,14 @@
+class Solution {
+ fun subarraySum(nums: IntArray, k: Int): Int {
+ val hm = hashMapOf(0 to 1)
+ var res = 0
+ var sum = 0
+ nums.forEach {
+ sum += it
+ val prefix = sum - k
+ res += hm.getOrDefault(prefix, 0)
+ hm[sum] = hm.getOrDefault(sum, 0) + 1
+ }
+ return res
+ }
+}