给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
用双指针减少for循环次数,将数组排序,让双指针可以有结束条件,也可以在有重复数字的时候直接跳过就好了。
python3代码
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
# 去掉特殊情况
length = len(nums)
if(not nums or length<3):
return []
# 先排序
nums.sort()
res=[]
for i in range(length):
# 设定返回条件
if(nums[i]>0):
return res
if(i>0 and nums[i]==nums[i-1]):
continue
L=i+1
R=length-1
while(L<R):
if(nums[i]+nums[L]+nums[R]==0):
res.append([nums[i],nums[L],nums[R]])
while(L<R and nums[L]==nums[L+1]):
L=L+1
while(L<R and nums[R]==nums[R-1]):
R=R-1
L=L+1
R=R-1
elif(nums[i]+nums[L]+nums[R]>0):
R=R-1
else:
L=L+1
return res

将代码写的更加清晰(TODO):
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
# 首先判断数组长度,如果小于三直接就返回空
length_of_nums = len(nums)
if length_of_nums < 3:
return []
# 长度大于等于三可以正常向下进行
# 首先要排好序,默认的升序排序
nums.sort()
# 接下来开始遍历排好序的数组,从头到尾
# 使用双指针进行指向性的数字操作
# 以遍历的值为初始值,指针永远向后移动,直到左指针和右指针相遇了结束
for i in nums:
# 设定左指针
L_needle = i + 1
# 设定右指针
R_needle = length_of_nums - 1
# 开始循环判断
# 为了规避重复值的问题,先比较i和左指针的值是否相同,再比较右指针和它前一项是否相同
# TODO: