双指针逼近--三数之和--(medium)

给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。

注意:答案中不可以包含重复的三元组。

示例:

给定数组 nums = [-1, 0, 1, 2, -1, -4],

满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:转换成俩数字之和求解,a1 = sum - a2; 前后逼近是值加起来等于a2

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        if len(nums) < 3:
            return []
        #排序
        nums.sort()
        arr = []
        # len(nums)-2 原因三数之和,要留一个凑3;如果是n-1,就没有后继,数量不能为3
        for i in range(0, len(nums)-2):  
            # 删除重复值
            if i> 0 and nums[i] == nums[i-1]:
                continue
            tmp = 0 - nums[i]
            j = i + 1
            k = len(nums) - 1
            while j < k:
                if nums[k] + nums[j] == tmp:
                    arr.append( [ nums[i], nums[j], nums[k] ] )
                    while j< k and nums[j] == nums[j+1]:
                        j += 1
                    while j< k and nums[k] == nums[k-1]:
                        k -= 1
                    j += 1
                    k -= 1 
                elif nums[k] + nums[j] > tmp:
                    k -= 1
                else:
                    j += 1
        return arr

你可能感兴趣的:(双指针逼近--三数之和--(medium))