📅  最后修改于: 2023-12-03 14:58:25.864000             🧑  作者: Mango
给定一个包含 $n$ 个整数的数组 $S$,和一个整数 $target$。
要求从 $S$ 中找到所有和为 $target$ 的不重复的三元组 $(a,b,c)$,并将它们按照元素从小到大的顺序输出。
注意,解决方案需要保证在输出结果中,不同的三元组不能重复。同一个三元组中的元素不能重复。
例如,对于数组 $S={-1,0,1,2,-1,-4}$,$target=0$,输出结果为 $[[-1, -1, 2], [-1, 0, 1]]$。
你需要实现以下函数:
def threeSum(nums: List[int], target: int) -> List[List[int]]:
pass
输入
输出
示例
输入:nums = [-1,0,1,2,-1,-4], target = 0
输出:[[-1,-1,2],[-1,0,1]]
这道题可以使用双指针进行解决,思路如下:
具体的程序实现如下:
from typing import List
def threeSum(nums: List[int], target: int) -> List[List[int]]:
ans = []
n = len(nums)
nums.sort()
for i in range(n):
if i > 0 and nums[i] == nums[i-1]: # 去重
continue
l, r = i + 1, n - 1
while l < r:
s = nums[i] + nums[l] + nums[r]
if s < target:
l += 1
elif s > target:
r -= 1
else:
ans.append([nums[i], nums[l], nums[r]])
while l < r and nums[l] == nums[l+1]: # 去重
l += 1
while l < r and nums[r] == nums[r-1]: # 去重
r -= 1
l += 1
r -= 1
return ans
测试一下我们的代码:
nums = [-1,0,1,2,-1,-4]
target = 0
print(threeSum(nums, target))
输出结果为:[[-1, -1, 2], [-1, 0, 1]]
,符合要求。