leetcode日记: 15. 三数之和

15. 三数之和

题目

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

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

示例

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

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

解法1–终极三重循环+去重

这题一看我以为是两数之和的翻版,不过hash我怎么也用不上.难受
暴力解法就是三重循环+去重

def threeSum(nums):
    result = []
    for i in range(len(nums)-2):
        a = nums[i]
        for j in range(len(nums)-i-1):
            b = nums[j+i+1]
            for x in range(len(nums)-i-j-2):
                c = nums[i+j+x+2]
                if a+b+c == 0:
                    temp = []
                    temp.append(a)
                    temp.append(b)
                    temp.append(c)
                    temp.sort()
                    if temp not in result:
                        result.append(temp)

    return result

解法2–固定一个+两头指针+左右移动法

先排序(太重要了)
固定一个的话其实还是三重循环,关键在于双指针一左一右向中间逼近
左指针的移动逻辑是顺序移动,重复的直接跳过.
右指针的移动逻辑(很重要)是abc求和,求和的结果大于0,右指针就左移一位.为啥呢?有序数组你往左移动一位才能减小嘛

def threeSum2(nums):
    if len(nums) < 3:
        return []
    nums.sort()
    result = []
    if nums[0] > 0:
        return result
    for index_a in range(len(nums)):
        if index_a > 0 and nums[index_a] == nums[index_a-1]:
            continue
        target = 0 - nums[index_a]
        index_c = len(nums) - 1
        for index_b in range(index_a+1, len(nums)):
            if index_b > index_a + 1 and nums[index_b] == nums[index_b - 1]:
                continue
            while index_b < index_c and nums[index_b] + nums[index_c] > target:
                index_c -= 1
            if index_b == index_c:
                break
            if nums[index_b] + nums[index_c] == target:
                result.append([nums[index_a], nums[index_b], nums[index_c]])
    return result

python学到了

今天没学到啥