Leecode 15. 三数之和

给你一个包含 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:





在《Leecode 15. 三数之和》上留下第一个评论