LeetCode #167 两数之和II - 输入有序数组 双指针

LeetCode #167 两数之和II - 输入有序数组

题目描述

给定一个已按照 升序排列 的有序数组，找到两个数使得它们相加之和等于目标数。

函数应该返回这两个下标值 index1 和 index2，其中 index1 必须小于 index2。

说明:

• 返回的下标值（index1 和 index2）不是从零开始的。
• 你可以假设每个输入只对应唯一的答案，而且你不可以重复使用相同的元素。
  示例:

输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。

方法一：一遍列表

立刻想到了 LeetCode #1 两数之和 哈希表 的解法，最近开始用 python 没想到用 HashMap 导致速度很慢

class Solution:
    def twoSum(self, numbers: List[int], target: int) -> List[int]:
    	if not numbers: return []
        l = {}
        for i in numbers:
            if target - i in l:
                return [l[target - i], len(l) + 1]
            else:
                l[i] = len(l) + 1
        
        return []

• 时间复杂度：$O(N)$
• 空间复杂度：$O(N)$

方法二：一遍哈希表

看到列表太慢了查发现 python 的字典就相当于 HashMap，但是这样写有问题，因为字典不可变，没有重复元素时可以，当出现相同元素时无法更新 value

class Solution:
    def twoSum(self, numbers: List[int], target: int) -> List[int]:
    	if not numbers: return []
        l = {}
        for i in numbers:
            if target - i in l:
                return [l[target - i] + 1, len(l) + 1]
            else:
                l[i] = len(l) + 1
        
        return []

• 时间复杂度：$O(N)$
• 空间复杂度：$O(N)$

方法三：双指针

这道题跟 LeetCode #1 两数之和 哈希表 很不同的特性就是数组已经排序。使用两个变量分别记录第一个元素和最后一个元素的位置，从首尾两边逼近，如果和大了说明右边的元素大了；和小了说明左边元素小了
图源 @CyC2018，侵删

class Solution:
    def twoSum(self, numbers: List[int], target: int) -> List[int]:
        if not numbers: return []

        i = 0; j = len(numbers) - 1
        while(i < j):
            ssum = numbers[i] + numbers[j]
            if(ssum == target):
                return [i+1, j+1]
            elif ssum < target:
                i += 1
            else:
                j -= 1
        
        return []

时间复杂度：$O(N)$
空间复杂度：$O(1)$：只需要两个变量记录位置
明明都是 $O(N)$ 法1做出来 756ms 法3 44ms。。。