python算法题库和答案_算法刷题(python)——三数之和

给定一个包含 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.sort() # 排序

res = []

i = 0

while i < len(nums) - 2: # i,j,k的位置[i, j,...,k]，jk可以相等，所以-2

if i == 0 or nums[i] != nums[i - 1]:

j = i + 1

k = len(nums) - 1

while j < k:

if nums[i] + nums[j] + nums[k] < 0:

j += 1 # <0要放大一点，所以j + 1

elif nums[i] + nums[j] + nums[k] > 0:

k -= 1

else:

res.append([nums[i], nums[j], nums[k]])

j += 1 # 只改一个数一定会不等于0

k -= 1

while j < k and nums[j] == nums[j - 1]:

j += 1 # 避免nums中的数重复，res中有想同的结果

while j < k and nums[k] == nums[k + 1]:

k -= 1

i += 1

return res