#241 3Sum - Dale 풀이 참고

Author: forest000014

3sum/forest000014.java

@@ -1,10 +1,10 @@
 /*
-# Time Complexity: O(n^2 * logn)
+# Time Complexity: O(n^2)
 # Space Complexity: O(1)
 */
 
 class Solution {
-    public List<List<Integer>> threeSum(int[] nums) {
+    public List<List<Integer>> threeSum1(int[] nums) { // solution 1
         int n = nums.length;
         Arrays.sort(nums);
         Set<List<Integer>> ans = new HashSet<>();
@@ -34,4 +34,30 @@ public List<List<Integer>> threeSum(int[] nums) {
 
         return new ArrayList<>(ans);
     }
+
+    public List<List<Integer>> threeSum(int[] nums) { // solution 2
+        int n = nums.length;
+        Arrays.sort(nums);
+        Set<List<Integer>> ans = new HashSet<>();
+
+        for (int i = 0; i < n - 2; i++) {
+            int l = i + 1;
+            int r = n - 1;
+            int target = -nums[i];
+            while (l < r) {
+                int sum = nums[l] + nums[r];
+                if (sum == target) {
+                    ans.add(new ArrayList<>(Arrays.asList(nums[i], nums[l], nums[r])));
+                    l++;
+                    r--; // 또 다른 (l, r) 조합이 있을 수 있으므로, loop를 계속 이어간다.
+                } else if (sum < target) {
+                    l++;
+                } else {
+                    r--;
+                }
+            }
+        }
+
+        return new ArrayList<>(ans);
+    }
 }