给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
这道题的前提是需要将数组变得有序,这样才能进行后序的工作。
思路:
nums[i]+nums[k] > target ---> k--
nums[i]+nums[k] == target ---> k--,i++
nums[i]+nums[k] < target ---> i++
nums.sort()
i=0
group = []
while (i <=len(nums)-2):
j=i+1
k=len(nums)-1
target = 0-nums[i]
while (j < k):
if nums[j]+nums[k] == target:
group.append([nums[i],nums[j],nums[k]])
j=j+1
k=k-1
while (nums[j] == nums[j-1] and j<k):
j=j+1
elif nums[j]+nums[k] > target:
k=k-1
while (nums[k] == nums[k+1] and j<k):
k=k-1
elif nums[j]+nums[k] < target:
j=j+1
while (nums[j] == nums[j-1] and j<k):
j=j+1
i=i+1
while (nums[i] == nums[i-1] and i < len(nums)-2):
i=i+1