[hackerrank] "Triple sum" 성능 개선 버전

Author: codehumane

src/main/kotlin/hackerrank/search/TripleSum.java

@@ -1,31 +1,47 @@
 package hackerrank.search;
 
 import java.util.Arrays;
-import java.util.HashSet;
-import java.util.Set;
 
 public class TripleSum {
 
     static long triplets(int[] a, int[] b, int[] c) {
 
-        // 예제로만 보면 주어진 배열들이 정렬되어 있음. 하지만 설명에는 명시되어 있지 않음.
         Arrays.sort(a);
         Arrays.sort(b);
         Arrays.sort(c);
 
+        long aCount = 0;
+        long cCount = 0;
+        int aIndex = 0;
+        int cIndex = 0;
         long count = 0L;
-        final Set<Integer> calculated = new HashSet<>();
+        int bCurrent = 0;
 
-        // 배열 안에 중복된 숫자가 있을 수 있음.
-        for (int number : b) {
+        for (int n : b) {
 
-            if (calculated.contains(number)) continue;
-            calculated.add(number);
+            // 같은 값이 존재할 수 있음.
+            if (n <= bCurrent) continue;
+            bCurrent = n;
 
-            final long left = Arrays.stream(a).filter(x -> x <= number).count();
-            final long right = Arrays.stream(c).filter(x -> x <= number).count();
-            count += left * right;
+            while (aIndex < a.length && a[aIndex] <= n) {
+                // 같은 값이 존재할 수 있음.
+                if (aIndex == 0 || a[aIndex - 1] != a[aIndex]) {
+                    aCount++;
+                }
 
+                aIndex++;
+            }
+
+            while (cIndex < c.length && c[cIndex] <= n) {
+                // 같은 값이 존재할 수 있음.
+                if (cIndex == 0 || c[cIndex - 1] != c[cIndex]) {
+                    cCount++;
+                }
+
+                cIndex++;
+            }
+
+            count += aCount * cCount;
         }
 
         return count;