算法题复健之路 第三天 数和

两数之和和三数之和 对应leetcode 1 和15

第一题简单，最容易想到的是暴力的检索遍历方法，这样时间复杂度是n^2，高级一点的做法也容易想到，字典，维护检索字典。

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        a = {}
        for i, num in enumerate(nums):
            if target-num in a:
                return [i, a[target-num]]
            a[nums[i]] = i

        return []

第二题较复杂，一开始我也想通过第一题的思路做，走了几个方法，暴力检索，分成正负，和维护一个查找数组皆超时了，原因在于这三种方法都需要遍历三次 也就是n^3的时间复杂度。

看了答案后领悟到，应该先排序，排序后通过双指针来检索，关键技巧在于比大小，通过先加跟0比来判断一个排好序的数组应该是左指针动还是右指针。这样最坏的情况只需要遍历两次，好的情况（不停跳过相同数字），则更快。

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort()
        res, k = [], 0
        for k in range(len(nums)-2):
            if nums[k] > 0: break
            if k > 0 and nums[k] == nums[k-1]: continue
            i, j = k+1, len(nums)-1
            while i < j:
                thesum = nums[k] + nums[i] + nums[j]
                if thesum < 0:
                    i+=1
                    while i < j and nums[i] == nums[i-1]: i+=1
                elif thesum > 0:
                    j-=1
                    while i < j and nums[j] == nums[j+1]: j-=1
                else:
                    res.append([nums[k], nums[i], 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

        return res