LeetCode 167. 两数之和 II - 输入有序数组（python解）

本文作为小白初学LeetCode笔记之用，如有不足之处欢迎指出

题目

在有序数组numbers中找出两个数，使其和为target（默认至少有一个解）

示例

输入：numbers = [2, 7, 11, 15] target = 9
输出：[1, 2]

自己的解1

class Solution:
    def twoSum(self, numbers: List[int], target: int) -> List[int]:
        lens = len(numbers)
        for p1 in range(lens-1):
            for p2 in range(p1+1, lens):
                if numbers[p1] + numbers[p2] == target:
                    return [p1+1, p2+1]

首先想到的是用一个两层循环来寻找解，时间复杂度为O(n²)，结果报错超出时间限制。

自己的解2

思考了一下报错的原因，上一个解的坏处是把所有组合的情况都考虑了进去，想到的优化方法是只考虑numbers[p1] + numbers[p2] < target 的情况。

class Solution:
    def twoSum(self, numbers: List[int], target: int) -> List[int]:
        lens = len(numbers)
        p1 = 0
        p2 = 1
        while p1 < lens-1 and numbers[p1] + numbers[p2] != target:
            if numbers[p1] + numbers[p2] > target:
                p1 += 1
                p2 = p1+1
            else:
                p2 += 1
                if p2 == lens:    # 如果p2超出数组范围
                    p1 += 1
                    p2 = p1+1 
        return [p1+1, p2+1]

时间复杂度仍是O(n²)，结果报错超出时间限制。我的思路一直是先固定第一个指针，然后向后移动第二个指针找满足答案的解，如果找不到则将第一个指针向后移动一位，再不断移动第二个指针进行尝试。这样始终没有摆脱复杂度是O(n²)的情况。

答案

思路：由于是有序数组，可将一个指针(left)指向较小的元素，将一个指针(right)指向较大的元素，left从左向右遍历，right从右向左遍历，令left与right指向的元素之和为sum。

如果 sum == target，那么得到要求的结果；
如果 sum > target，移动较大的元素，使 sum 变小一些；
如果 sum < target，移动较小的元素，使 sum 变大一些。

class Solution:
    def twoSum(self, numbers: List[int], target: int) -> List[int]:
        lens = len(numbers)
        left = 0
        right = lens-1
        while left < right:
            sum = numbers[left] + numbers[right]
            if sum == target:
                return [left+1, right+1]
            elif sum < target:
                left += 1
            else:
                right -= 1
        return None

这样做的好处是两个指针分别从两侧向中间靠近，使得整个数组只遍历了一次，时间复杂度为O(n)。

总结

在有序数组中，要学会使用一前一后的双指针，不要只想着两个指针都从从前往后遍历。

参考

[1] leetcode官方题解
[2] CyC2018 Leetcode 题解 - 双指针