Create Number of Arithmetic Triplets.java

varunu28 · web-flow · commit 2c654bd46ae6 · 2022-08-07T08:09:12.000-07:00
diff --git a/Easy/Number of Arithmetic Triplets.java b/Easy/Number of Arithmetic Triplets.java
@@ -0,0 +1,13 @@
+class Solution {
+  public int arithmeticTriplets(int[] nums, int diff) {
+    int count = 0;
+    Set<Integer> set = new HashSet<>();
+    for (int num : nums) {
+      if (set.contains(num - diff) && set.contains(num - 2 * diff)) {
+        count++;
+      }
+      set.add(num);
+    }
+    return count;
+  }
+}
