Update beautifulTriplet.py

Ansumanbhujabal · Feb 28, 2024 · 3c2a97b · 3c2a97b
1 parent 27b21c6
commit 3c2a97b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/Python_Problems/beautifulTriplet.py b/Python_Problems/beautifulTriplet.py
@@ -19,4 +19,12 @@ def beautifulTriplets(d, arr):
     return len(total)//3
 
 
+# Optimized
+
+def beautifulTriplets(d, arr):
+    total = sum(1 for i in arr if i+d in arr and i+2 *
+                d in arr)  # counts 1 for each element
+    return total
+
+
 print(beautifulTriplets(1, [2, 2, 3, 4, 5]))