leetcode第15题三数之和--双指针

最新推荐文章于 2024-09-09 18:27:53 发布

lxhseph

最新推荐文章于 2024-09-09 18:27:53 发布

阅读量246

点赞数

文章标签： leetcode 算法 python

本文链接：https://blog.csdn.net/lxhseph/article/details/122478829

版权

先记一下思路上的错误，一开始受了第11题的影响想着双指针就是从头尾开始，第三个指针在头尾中间：

如果中间指针在最左边且三数之和大于零（最小三数之和大于零），则右指针过大，右指针左移；如果中间指针在最右边且三数之和小于零（最大三数之和小于零），则左指针过小，左指针右移；但是有问题就是其他情况的话该怎么办？这里没有想到办法

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        思路：这个用双指针怎么弄啊，想想哈
        """
        if len(nums) <= 2: return []
        nums.sort()
        list_return = []
        i = 0
        j = len(nums) - 1
        while i < j:
            if nums[i] > 0 or nums[j] < 0:#todo i或者j已经越界
                return list_return
            elif nums[i]+nums[i+1]+nums[j]>0:#todo i过小，右移
                j-=1
            elif nums[i]+nums[j-1]+nums[j]<0:#todo j过大，左移
                i+=1
            elif i!=0:
                if nums[i] == nums[i - 1]:
                    i+=1
            elif j!=len(nums)-1:
                if nums[j] == nums[j + 1]:
                    j-=1
            for k in range(i + 1, j):
                if k!=i+1:
                    if nums[k]==nums[k-1]:
                        break
                if nums[i]+nums[j]+nums[k] == 0:
                    list_return.append([nums[i],nums[k],nums[j]])
            i+=1
        return list_return

官方的题解是对于每一个i（0~len（nums）-1），三个指针起始位置分别是：i，i+1，len（nums）-1，当j<k时，若nums[i]+nums[j]+nums[k]>0，k左移，若nums[i]+nums[j]+nums[k]<0，j右移。记得去重

class Solution(object):
    def threeSum(self, nums):
        if len(nums) <= 2: return []
        nums.sort()
        list_return = []
        for i in range(len(nums)-1):
            if nums[i]>0:
                return list_return
            if i!=0:
                if nums[i]==nums[i-1]:
                    continue
            j = i + 1
            k = len(nums) - 1
            while j < k:
                if j != i + 1:
                    if nums[j] == nums[j - 1]:
                        j += 1
                        continue
                if k != len(nums) - 1:
                    if nums[k] == nums[k + 1]:
                        k -= 1
                        continue
                sum_temp = nums[i] + nums[j] + nums[k]
                if sum_temp > 0:
                    k -= 1
                elif sum_temp < 0:
                    j += 1
                else:
                    list_return.append([nums[i], nums[j], nums[k]])
                    j += 1
                    k -= 1
        return list_return

记得continue的用法，跳过本次循环