LC15. 三数之和（排序+双指针-中等）

LC15. 三数之和

题目：给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。
思路：
        ① 暴力    时间复杂度为O(n^3)
        ② 排序（nlogn） + 双指针(一固定n * 双指针 n)
            [-1 0 1 2 -1 -4]  排序后   [-4 -1 -1 0 1 2]
            i 固定为最左侧一个数（外层循环中）  j k 使用双指针进行遍历
            PS：
            外层遍历i：  当 i 的下几个和当前i对应的值相同  则略过
            双指针遍历时（jk）：
                   ① 和 等于 0    添加对应元素并增j减k 并过滤掉和当前jk相同的值  如 -4 1 1 1 2 2 3 3 3
                   ② 和 小于 0    增j 并过滤掉和当前j相同的值 如 -8 -5 -5 -5 1 1 1 2 2 3
                   ③ 和 大于 0    减k 并过滤掉和当前k相同的值  如 -8 0 0 0 0  1 2 2 3 3 3
复杂度： 时间复杂度 O(n^2)  空间复杂度 O(1)

代码：

'''
    题目：给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。
    思路：
        ① 暴力    时间复杂度为O(n^3)
        ② 排序（nlogn） + 双指针(一固定n * 双指针 n)
            [-1 0 1 2 -1 -4]  排序后   [-4 -1 -1 0 1 2]
            i 固定为最左侧一个数（外层循环中）  j k 使用双指针进行遍历
            PS：
            外层遍历i：  当 i 的下几个和当前i对应的值相同  则略过
            双指针遍历时（jk）：
                        ① 和 等于 0    添加对应元素并增j减k 并过滤掉和当前jk相同的值  如 -4 1 1 1 2 2 3 3 3
                        ② 和 小于 0    增j 并过滤掉和当前j相同的值 如 -8 -5 -5 -5 1 1 1 2 2 3
                        ③ 和 大于 0    减k 并过滤掉和当前k相同的值  如 -8 0 0 0 0  1 2 2 3 3 3
    复杂度： 时间复杂度 O(n^2)  空间复杂度 O(1)
'''

def threeSum(nums):
    n = len(nums)
    if n < 3:
        return []
    res = []
    nums.sort()

    for i in range(n-2):
        # 当最小的元素大于0  则和一定大于0
        if nums[i] > 0:
            break

        # 去第一个的元素重复
        if i >= 1 and nums[i] == nums[i-1]:
            continue

        j, k = i + 1, n-1
        while j < k:
            # 对于 jk也要去除重复元素  不同的是  务必保证判断了之和再去除   不然类似 2 =1 + 1 会被过滤掉

            s = nums[i] + nums[j] + nums[k]
            if s == 0:
                res.append([nums[i], nums[j], nums[k]])
                j += 1
                k -= 1
                while j < k and nums[j] == nums[j - 1]: j += 1
                while j < k and nums[k] == nums[k + 1]: k -= 1
            elif s < 0:
                j += 1
                while j < k and nums[j] == nums[j-1]: j += 1
            else:
                k -= 1
                while j < k and nums[k] == nums[k + 1]: k -= 1
    return res

numsold = [-4,-2,1,-5,-4,-4,4,-2,0,4,0,-2,3,1,-5,0]
print(threeSum(numsold))

路虽远，行则将至。事虽难，做则必成