leetcode.com 15. 3Sum v2

문제 링크: https://leetcode.com/problems/3sum/

Author: skysign

leetcode.com 15. 3Sum v2/main.py

@@ -0,0 +1,24 @@
+from typing import List
+
+
+class Solution:
+    def threeSum(self, nums: List[int]) -> List[List[int]]:
+        nums.sort()
+        myset = set()
+
+        for idx_left in range(0, len(nums) - 2):
+            target = -nums[idx_left]
+            idx_mid = idx_left + 1
+            idx_right = len(nums) - 1
+
+            while idx_mid < idx_right:
+                if nums[idx_mid] + nums[idx_right] == target:
+                    myset.add((nums[idx_left], nums[idx_mid], nums[idx_right]))
+                    idx_mid += 1
+                    idx_right -= 1
+                elif nums[idx_mid] + nums[idx_right] < target:
+                    idx_mid += 1
+                elif nums[idx_mid] + nums[idx_right] > target:
+                    idx_right -= 1
+
+        return [list(item) for item in list(myset)]

leetcode.com 15. 3Sum v2/test_main.py

@@ -0,0 +1,20 @@
+from unittest import TestCase
+from main import Solution
+
+
+class TestSolution(TestCase):
+    def test_three_sum(self):
+        sln = Solution()
+        self.assertEqual(sln.threeSum([-1,0,1,2,-1,-4]), [[-1,0,1], [-1,-1,2]])
+
+    def test_three_sum2(self):
+        sln = Solution()
+        self.assertEqual(sln.threeSum([0, 1, 1]), [])
+
+    def test_three_sum3(self):
+        sln = Solution()
+        self.assertEqual(sln.threeSum([0, 0, 0]), [[0, 0, 0]])
+
+    def test_three_sum4(self):
+        sln = Solution()
+        self.assertEqual(sln.threeSum([1,-1,-1,0]), [[-1, 0, 1]])