三数之和
题目:给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,
使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[ [-1, 0, 1],[-1, -1, 2]]
解法:利用双指针法思路
1.当 nums[k] > 0 时直接break跳出:因为 nums[j] >= nums[i] >= nums[k] > 0,即 3 个数字都大于 0 ,在此固定指针 k 之后不可能再找到结果了。
2.先将数组nums排序,排序后图片如下图,然后分别设置三个指针k,i,j, k为长度为数组索引的循环,从0到(len(nums)-1),i则为k+1,j为(len(nums)-1)
3.k为0时,当s=nums[k] + nums[i] + nums[j] < 0时,i值加1,并判断此时的nums[i]是否和nums[i-1]相等,如果相等,则i再加1,直到不相等或i大于j为止。当s=nums[k] + nums[i] + nums[j] > 0时,j值减1,并判断此时的nums[j]是否和nums[j+1]相等,如果相等,则j再减1,直到不相等或i大于j为止。当s=nums[k] + nums[i] + nums[j] == 0时,返回值[nums[k],nums[i] ,nums[j]],并分别将i加1,j减1,并跳过所有重复的nums[i]和nums[j],防止记录到重复组合。
4.当i与j相等时,k值加1,然后按照上面方法继续计算,代码如下:
class Solution(object):
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
nums.sort()
res = []
for k in range(len(nums)-2):
if nums[k] > 0:
break
if k > 0 and nums[k] == nums[k-1]:
continue
i = k+1
j = len(nums)-1
while(i<j):
s = nums[k] + nums[i] + nums[j]
if s > 0:
j -= 1
while i < j and nums[j] == nums[j+1]:
j -= 1
elif s < 0:
i += 1
while i < j and nums[i] == nums[i-1]:
i += 1
else:
res.append([nums[k],nums[i],nums[j]])
i += 1
j -= 1
while i < j and nums[j] == nums[j + 1]:
j -= 1
while i < j and nums[i] == nums[i - 1]:
i += 1
return res