三数之和(threeSum)*

15.三数之和

https://leetcode-cn.com/problems/3sum/
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。

示例:
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],

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

解析:
1.暴力解:
遍历数组,列举所有三三组合,判断是否等于 target。复杂度 O(n^3)
2.排序+双指针 复杂度O(n^2)

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        res = []
        nums.sort()
        for k in range(len(nums)-2):
            # nums[k] > 0 则无解
            if nums[k] > 0:
                break
            # 跳过重复的k, 注意如果没有第一个条件则无法正确处理 nums = [0,0,0,0] 的情况
            if k > 0 and nums[k] == nums[k-1]:
                continue
            #双指针
            i = k + 1
            j = len(nums) - 1
            while i < j:
                s = nums[k] + nums[i] + nums[j]
                if s == 0:
                    res.append([nums[i] , nums[k] , nums[j]])
                    i += 1
                    j -= 1
                    # 跳过重复
                    while i < j and nums[i] == nums[i-1]:
                        i += 1
                    while i < j and nums[j] == nums[j+1]:
                        j -= 1
                if s < 0:
                    i += 1
                    while i < j and nums[i] == nums[i-1]:
                        i += 1
                if s > 0:
                    j -= 1
                    while i < j and nums[j] == nums[j+1]:
                        j -= 1
        return res

3.知识点:
排序复杂度是O(nlogn).
双指针开始一头一尾
注意跳过重复的值

你可能感兴趣的:(三数之和(threeSum)*)