给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
解题思路:
排序+双指针
先从小到大排列,每次取出一个数之后对于剩下的数采用两数之和的方法(双指针)进行查找
注意:重点是如何高效的去除重复元素
将数组排序之后,在进行查找的时候需要将重复的数字跳过去(防止出现重复结果)即可
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
nums=sorted(nums)#从小到大排列
res=[]#存放返回结果
for num in range(len(nums)-2):
x=0-nums[num]
if num==0 or nums[num]>nums[num-1]:
i,j=num+1,len(nums)-1
while i<j:
if nums[i]+nums[j]==x:
res.append([nums[num],nums[i],nums[