Subarray sum equals k

Author: nlohia-godaddy

src/main/java/leetcode/medium/SubArraySumEqualsK.java

@@ -9,7 +9,7 @@
 
 class SubArraySumEqualsK {
 
-  public int subarraySum(int[] nums, int k) {
+  int subarraySum(int[] nums, int k) {
 
     Map<Integer, Integer> sumCountMap = new HashMap<>();
     sumCountMap.put(0, 1);
@@ -22,7 +22,9 @@ public int subarraySum(int[] nums, int k) {
       if (sumCountMap.containsKey(prefixSum - k)) {
         result += sumCountMap.get(prefixSum - k);
       }
-      sumCountMap.put(prefixSum, sumCountMap.getOrDefault(prefixSum, 0) + 1);
+
+      sumCountMap.put(prefixSum,
+          sumCountMap.getOrDefault(prefixSum, 0) + 1);
     }
 
     return result;

src/test/java/leetcode/medium/SubArraySumEqualsKTest.java

@@ -0,0 +1,38 @@
+package leetcode.medium;
+
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+class SubArraySumEqualsKTest {
+
+  private final SubArraySumEqualsK subArraySumEqualsK;
+
+  SubArraySumEqualsKTest() {
+    subArraySumEqualsK = new SubArraySumEqualsK();
+  }
+
+  @Test
+  void testSubarraySum1() {
+    int[] nums = {1, 1, 1};
+    int k = 2;
+
+    assertEquals(2, subArraySumEqualsK.subarraySum(nums, k));
+  }
+
+  @Test
+  void testSubarraySum2() {
+    int[] nums = {1, 2, 3};
+    int k = 3;
+
+    assertEquals(2, subArraySumEqualsK.subarraySum(nums, k));
+  }
+
+  @Test
+  void testSubarraySum3() {
+    int[] nums = {1, -1, 0};
+    int k = 0;
+
+    assertEquals(3, subArraySumEqualsK.subarraySum(nums, k));
+  }
+}