题目描述请点击查看LeetCode 题目描述
Python3 代码解答如下:
class Solution:
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
"""
思路:先将数组排序,后循环数组,假定第 i 个数为第一个元素,第二个元素为第i+1个,第3个元素为数组最后一个,因为前面的都是数值较小的,要想找到与其相反的大数,从后面开始会比较快,因为不能重复,所以要保证第二个元素下标要小于第3个元素
"""
nums.sort()
res =[]
i = 0
for i in range(len(nums)):
#当第一次循环的第一个元素与第二次循环的第一个元素相等时,不必找了,即使找到也都是重复的了
if i == 0 or nums[i]>nums[i-1]:
l = i+1
r = len(nums)-1
#限制第二个元素与第三个元素的重复使用,比如2,3 = 3,2
while l < r:
s = nums[i] + nums[l] +nums[r]
if s ==0:
#找到元素,追加到列表中,并第二个元素加1,第三个元素减1,两个只要有一个不变,即使有满足条件的元素,也会重复
res.append([nums[i],nums[l],nums[r]])
l +=1
r -=1
#判断下标+1后的第二个元素与第二个元素是否一样,如果一样依然是+1
while l < r and nums[l] == nums[l-1]:
l += 1
#判断下标-1后的第三个元素与第三个元素是否一样,如果一样依然是-1
while r > l and nums[r] == nums[r+1]:
r -= 1
elif s>0:
#当和大于0时,说明第三个元素的值太大,需要一个稍小一点的,所有下标-1
r -=1
else :
#当和小于0时,说明前两个元素的值太小,需要一个稍大一点的,因为第一个元素固定为第 i个,所以第二元素下标+1
l +=1
return res