class Solution(object):
def inn(self, nums):
n = len(nums)
if (not nums or n <= 2):
return []
nums.sort()
result = []
for i in range(n):
if (nums[i] > 0):
return result
if (i > 0 and nums[i] == nums[i - 1]):
continue
head = i + 1
tail = n - 1
while (head < tail):
if (nums[i] + nums[head] + nums[tail] == 0):
result.append([nums[i], nums[head], nums[tail]])
while (head < tail and nums[head] == nums[head + 1]):
head = head + 1
while (head < tail and nums[tail] == nums[tail - 1]):
tail = tail - 1
head = head + 1
tail = tail - 1
elif (nums[i] + nums[head] + nums[tail] > 0):
tail = tail - 1
else:
head = head + 1
return result
思路:排序+双指针
题目中要求:答案中不可以包含重复的三元组
首先,什么算是重复?对于数组[-1,-1,0,-4,2,1]来说,会有相同的解,即:[-1,-1,2],-1可以是第一个-1,也可以是第二个-1,所以重复。
之后,对于该题目,可以将三个数简化为如下关系式:
head+tail=-i
而对于满足不重复的要求,可以先将数组排序,让i<head<tail
(1)对数组排序
(2)让i遍历数组:
让head处于i+1的位置,tail处于数组尾
如果i所指的元素>0,则head,tail都>0,固不存在解
如果nums[head]+nums[tail]<nums[i],则移动head,head=head+1
如果nums[head]+nums[tail]>nums[i],则移动tail,tail=tail-1
如果相等,则同时移动head和tail,寻找下一个解。