LeetCode167两数之和2输入有序数组

发表于 | 分类于 |

题目描述

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

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

说明:

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

示例:

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

解题思路

审题可知，有序的数组，而且不存在重复的元素（只对应唯一的答案），适合使用双数组解题。

使用start和end两个指针分别从数组的头部和尾部遍历数组，比较 numbers[start] + numbers[end] 与 tartget 的大小，来移动左右指针。

代码实现

GitHub Python代码地址

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
class Solution(object):
    def twoSum(self, numbers, target):
        """
        :type numbers: List[int]
        :type target: int
        :rtype: List[int]
        """
        start = 0
        end = len(numbers) - 1

        while start < end:
            result = numbers[start] + numbers[end]
            if result == target:
                return [start+1, end+1]
            elif result > target:
                end = end - 1
            else:
                start = start + 1