解题思路-leetcode第十五题:三数之和
题目描述:给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
本次采用先排序再双指针遍历的方法,第一步是对特殊输入的处理,比如输入为空或长度小于3,则返回空列表。然后对列表进行升序排列,排序后遍历数组元素,并设置两个指针L,R,L指向遍历元素的下一元素,R指向最后,当L<R时,进入循环,若此时遍历位置元素加两指针元素之和等于0,则返回此三元素组成的列表,然后左指针右移,右指针左移,若三元素之和大于0,则右指针左移,若小于0,则左指针右移。但此时并未考虑如何去重,所以要做去重处理,首先在遍历元素时,若元素值大于0,则返回此时的列表,若当前元素等于上一元素,则跳出此次循环,然后是在while循环内,在三数之和等于0的条件下,若L<R且左指针元素等于左指针下一元素,则左指针右移,若L<R且右指针元素等于右指针前一元素,则右指针左移。代码如下:
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
sum = []
if(not nums or len(nums)<3):
return []
nums.sort()
for i in range(len(nums)):
if(nums[i]>0):
return sum
if(i>0 and nums[i]==nums[i-1]):
continue
L=i+1
R=len(nums)-1
while(L<R):
if(nums[i]+nums[L]+nums[R]==0):
sum.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 sum
提交后,通过。
本文详细解析了LeetCode第15题“三数之和”的解题思路,采用先排序再双指针遍历的方法,避免重复解并提供完整Python代码实现。
1万+

被折叠的 条评论
为什么被折叠?



