|
1 | 1 | class Solution: |
2 | | - def ThreeSum(self, integers): |
3 | | - """ |
4 | | - :type integers: List[int] |
5 | | - :rtype: List[List[int]] |
6 | | - """ |
7 | | - integers.sort() |
8 | | - result = [] |
9 | | - for index in range(len(integers)): |
10 | | - if integers[index] > 0: |
| 2 | + def threeSum(self, nums: List[int]) -> List[List[int]]: |
| 3 | + res = [] |
| 4 | + nums.sort() |
| 5 | + |
| 6 | + for i, a in enumerate(nums): |
| 7 | + # Skip positive integers |
| 8 | + if a > 0: |
11 | 9 | break |
12 | | - if index > 0 and integers[index] == integers[index - 1]: |
| 10 | + |
| 11 | + if i > 0 and a == nums[i - 1]: |
13 | 12 | continue |
14 | | - left, right = index + 1, len(integers) - 1 |
15 | | - while left < right: |
16 | | - if integers[left] + integers[right] < 0 - integers[index]: |
17 | | - left += 1 |
18 | | - elif integers[left] + integers[right] > 0 - integers[index]: |
19 | | - right -= 1 |
| 13 | + |
| 14 | + l, r = i + 1, len(nums) - 1 |
| 15 | + while l < r: |
| 16 | + threeSum = a + nums[l] + nums[r] |
| 17 | + if threeSum > 0: |
| 18 | + r -= 1 |
| 19 | + elif threeSum < 0: |
| 20 | + l += 1 |
20 | 21 | else: |
21 | | - result.append([integers[index], integers[left], integers[right]]) # After a triplet is appended, we try our best to incease the numeric value of its first element or that of its second. |
22 | | - left += 1 # The other pairs and the one we were just looking at are either duplicates or smaller than the target. |
23 | | - right -= 1 # The other pairs are either duplicates or greater than the target. |
24 | | - # We must move on if there is less than or equal to one integer in between the two integers. |
25 | | - while integers[left] == integers[left - 1] and left < right: |
26 | | - left += 1 # The pairs are either duplicates or smaller than the target. |
27 | | - return result |
| 22 | + res.append([a, nums[l], nums[r]]) |
| 23 | + l += 1 |
| 24 | + r -= 1 |
| 25 | + while nums[l] == nums[l - 1] and l < r: |
| 26 | + l += 1 |
| 27 | + |
| 28 | + return res |
0 commit comments