LeetCode 167-两数之和II

116 阅读2分钟

题目

给你一个下标从 1 开始的整数数组 numbers ,该数组已按 ****非递减顺序排列 ** ,请你从数组中找出满足相加之和等于目标数 target 的两个数。如果设这两个数分别是 numbers[index1] 和 numbers[index2] ,则 1 <= index1 < index2 <= numbers.length 。

以长度为 2 的整数数组 [index1, index2] 的形式返回这两个整数的下标 index1 **和 **index2

你可以假设每个输入 只对应唯一的答案 ,而且你 不可以 重复使用相同的元素。

你所设计的解决方案必须只使用常量级的额外空间。

  示例 1:

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

示例 2:

输入: numbers = [2,3,4], target = 6
输出: [1,3]
解释: 2 与 4 之和等于目标数 6 。因此 index1 = 1, index2 = 3 。返回 [1, 3] 。

示例 3:

输入: numbers = [-1,0], target = -1
输出: [1,2]
解释: -1 与 0 之和等于目标数 -1 。因此 index1 = 1, index2 = 2 。返回 [1, 2] 。

解法一:暴力for循环(略)

解法二:双指针

思路如下

  • 初始时两个指针分别指向第一个元素位置和最后一个元素的位置。
  • 每次计算两个指针指向的两个元素之和,并和目标值比较。
    • 如果两个元素之和等于目标值,则发现了唯一解。
    • 如果两个元素之和小于目标值,则将左侧指针右移一位。
    • 如果两个元素之和大于目标值,则将右侧指针左移一位。
  • 移动指针之后,重复上述操作,直到找到答案。

小小优化

在移动指针时,可以多次判读那是否一直不满足条件,然后将指针继续移动,直到满足条件位置。

代码

class Solution:
    def twoSum(self, numbers: List[int], target: int) -> List[int]:
        i = 0
        j = len(numbers) - 1
        while i < j:
            s = numbers[i] + numbers[j]
            if s == target:
                return [i+1, j+1]
            elif s > target:
                while i < j and numbers[j-1] + numbers[i] > target:
                    j -= 1
                j -= 1
            else:
                while i < j and numbers[i+1] + numbers[j] < target:
                    i += 1
                i += 1