关键在去重,而且题目要求返回三个值,所以使用排序 + 双指针
记住这里不要求返回具体值索引,所以排序直接使用nums.sore()即可,节约空间;若需要保留原列表可以使用new_nums = sorted(nums).
python
class Solution:
def threeSum(self, nums: list[int]) -> list[list[int]]:
if not nums:
return []
nums.sort()
length = len(nums)
res = []
for i in range(length):
# 首位去重
if i > 0 and nums[i] == nums[i-1]:
continue
left = i + 1
right = length -1
while left < right:
sum = nums[i] + nums[left] + nums[right]
if sum == 0:
res.append([nums[i], nums[left], nums[right]])
# left标志位去重
while left < right and nums[left] == nums[left + 1]:
left += 1
# right 标志位去重
while left < right and nums[right] == nums[right -1]:
right -= 1
left += 1
right -= 1
elif sum < 0:
left += 1
else:
right -= 1
return res