WIP

Author: Gonzalo Diaz

algorithm-exercises-java/src/main/java/ae/hackerrank/interview_preparation_kit/dictionaries_and_hashmaps/CountTriplets.java

@@ -0,0 +1,54 @@
+package ae.hackerrank.interview_preparation_kit.dictionaries_and_hashmaps;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * CountTriplets.
+ *
+ * @link Problem definition
+ *       [[docs/hackerrank/interview_preparation_kit/dictionaries_and_hashmaps/two-strings.md]]
+ * @link Solution notes
+ *       [[docs/hackerrank/interview_preparation_kit/dictionaries_and_hashmaps/two-strings-solution-notes.md]]
+ */
+public class CountTriplets {
+  private CountTriplets() {
+  }
+
+  /**
+   * CountTriplets.
+   */
+  static long countTriplets(List<Long> arr, long r) {
+
+    Map<Long, Integer> a_counter = new HashMap<>();
+    Map<Long, Integer> b_counter = new HashMap<>();
+    Long triplets = 0L;
+
+    for (Long item : arr) {
+      if (a_counter.putIfAbsent(item, 1) != null) {
+        Integer currentValue = a_counter.get(item);
+        a_counter.put(item, currentValue + 1);
+      }
+    }
+
+    for (Long item : arr) {
+      Long j = item / r;
+      Long k = item * r;
+      a_counter.put(item, a_counter.get(item) - 1);
+      if (b_counter.containsKey(j) && a_counter.containsKey(k) && item % r == 0) {
+        triplets += b_counter.get(j) * a_counter.get(k);
+      }
+
+      Integer b_item = b_counter.get(item);
+
+      if (b_item != null) {
+        b_counter.put(item, b_item + 1);
+      } else {
+        b_counter.put(item, 1);
+      }
+    }
+
+    return triplets;
+  }
+}

algorithm-exercises-java/src/test/java/ae/hackerrank/interview_preparation_kit/dictionaries_and_hashmaps/CountTripletsTest.java

@@ -0,0 +1,76 @@
+package ae.hackerrank.interview_preparation_kit.dictionaries_and_hashmaps;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import java.io.IOException;
+import java.util.List;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.TestInstance;
+import org.junit.jupiter.api.TestInstance.Lifecycle;
+import util.JsonLoader;
+
+/**
+ * CountTripletsTest.
+ */
+@TestInstance(Lifecycle.PER_CLASS)
+class CountTripletsTest {
+  public static class CountTripletsTestCase {
+    public String title;
+    public List<Long> input;
+    public Long r;
+    public Long expected;
+  }
+
+  private List<CountTripletsTestCase> smallTestCases;
+  private List<CountTripletsTestCase> bigTestCases;
+
+  @BeforeAll
+  public void setup() throws IOException {
+    String path;
+    path = String.join("/",
+        "hackerrank",
+        "interview_preparation_kit",
+        "dictionaries_and_hashmaps",
+        "count_triplets_1.small.testcases.json");
+
+    this.smallTestCases = JsonLoader.loadJson(path, CountTripletsTestCase.class);
+
+    path = String.join("/",
+        "hackerrank",
+        "interview_preparation_kit",
+        "dictionaries_and_hashmaps",
+        "count_triplets_1.big.testcases.json");
+
+    this.bigTestCases = JsonLoader.loadJson(path, CountTripletsTestCase.class);
+
+  }
+
+  @Test
+  void testCountTriplets() {
+    for (CountTripletsTestCase test : smallTestCases) {
+      Long solutionFound = CountTriplets.countTriplets(test.input, test.r);
+
+      assertEquals(test.expected, solutionFound,
+          "%s(%s, %d) answer must be: %d".formatted(
+              "CountTriplets.countTriplets",
+              test.input.toString(),
+              test.r,
+              test.expected));
+    }
+  }
+
+  @Test
+  void testCountTripletsBigCases() {
+    for (CountTripletsTestCase test : bigTestCases) {
+      Long solutionFound = CountTriplets.countTriplets(test.input, test.r);
+
+      assertEquals(test.expected, solutionFound,
+          "%s(%s, %d) answer must be: %d".formatted(
+              "CountTriplets.countTriplets",
+              test.input.toString(),
+              test.r,
+              test.expected));
+    }
+  }
+}

algorithm-exercises-java/src/test/resources/hackerrank/interview_preparation_kit/dictionaries_and_hashmaps/count_triplets_1.big.testcases.json

@@ -0,0 +1,26 @@
+[
+  {
+    "title": "Sample Test Case 0",
+    "input": [1, 2, 2, 4],
+    "r": 2,
+    "expected": 2
+  },
+  {
+    "title": "Sample Test Case 1",
+    "input": [1, 3, 9, 9, 27, 81],
+    "r": 3,
+    "expected": 6
+  },
+  {
+    "title": "Sample Test Case 1 (unsorted)",
+    "input": [9, 3, 1, 81, 9, 27],
+    "r": 3,
+    "expected": 1
+  },
+  {
+    "title": "Sample Test Case 12",
+    "input": [1, 5, 5, 25, 125],
+    "r": 5,
+    "expected": 4
+  }
+]