Leetcode_167 Two Sum II - Input array is sorted

发表于 | 分类于

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

函数应该返回这两个下标值 index1 和 index2，其中 index1 必须小于 index2。请注意，返回的下标值（index1 和 index2）不是从零开始的。

你可以假设每个输入都只有一个解决方案，而且你不会重复使用相同的元素。

输入：数组 = {2, 7, 11, 15}, 目标数 = 9
输出：index1 = 1, index2 = 2

“””
采用二分法:因为有序,首位相加和目标值判断：

  1. 大于目标值，移动尾部到前一个

  2. 小于目标值，移动首部到下一个

  3. 相等返回，要求索引从1开始，所以返回的是得到的位置加1

“””

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
class Solution:
    def twoSum(self, numbers, target):
        """
        :type numbers: List[int]
        :type target: int
        :rtype: List[int]
        """
        low, high = 0, len(numbers) - 1
        while low < high:
            tmp = numbers[low] + numbers[high]
            if tmp == target:
                return [low + 1, high + 1]
            elif tmp > target:
                high -= 1
            else:
                low += 1