15. 三数之和
题目
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
解法1–终极三重循环+去重
这题一看我以为是两数之和的翻版,不过hash我怎么也用不上.难受
暴力解法就是三重循环+去重
def threeSum(nums):
result = []
for i in range(len(nums)-2):
a = nums[i]
for j in range(len(nums)-i-1):
b = nums[j+i+1]
for x in range(len(nums)-i-j-2):
c = nums[i+j+x+2]
if a+b+c == 0:
temp = []
temp.append(a)
temp.append(b)
temp.append(c)
temp.sort()
if temp not in result:
result.append(temp)
return result
解法2–固定一个+两头指针+左右移动法
先排序(太重要了)
固定一个的话其实还是三重循环,关键在于双指针一左一右向中间逼近
左指针的移动逻辑是顺序移动,重复的直接跳过.
右指针的移动逻辑(很重要)是abc求和,求和的结果大于0,右指针就左移一位.为啥呢?有序数组你往左移动一位才能减小嘛
def threeSum2(nums):
if len(nums) < 3:
return []
nums.sort()
result = []
if nums[0] > 0:
return result
for index_a in range(len(nums)):
if index_a > 0 and nums[index_a] == nums[index_a-1]:
continue
target = 0 - nums[index_a]
index_c = len(nums) - 1
for index_b in range(index_a+1, len(nums)):
if index_b > index_a + 1 and nums[index_b] == nums[index_b - 1]:
continue
while index_b < index_c and nums[index_b] + nums[index_c] > target:
index_c -= 1
if index_b == index_c:
break
if nums[index_b] + nums[index_c] == target:
result.append([nums[index_a], nums[index_b], nums[index_c]])
return result
python学到了
今天没学到啥