剑指 Offer 57. 和为s的两个数字

题目描述

输入一个递增排序的数组和一个数字s，在数组中查找两个数，使得它们的和正好是s。如果有多对数字的和等于s，则输出任意一对即可。

示例 1：
输入：nums = [2,7,11,15], target = 9
输出：[2,7] 或者 [7,2]

示例 2：
输入：nums = [10,26,30,31,47,60], target = 40
输出：[10,30] 或者 [30,10]

解题思路: 头尾指针法

1. 这个题目与第一题类似, 都是查找target的和, 所以我们可以使用第一题的map法进行解题
2. 同时, 因为本题是递增的数组, 所以我们可以通过头尾双指针法进行遍历
3. 使两个指针分别对应头尾, 然后判断2个位置元素的和与target, 如果大于target, 那么就让尾指针向前移动, 如果小于target, 就让头指针向后移动, 知道找到target

示例代码

def twoSum(self, nums: List[int], target: int) -> List[int]:
    l, r = 0, len(nums) - 1

    while l < r:
        if nums[l] + nums[r] < target:
            l += 1
        elif nums[l] + nums[r] > target:
            r -= 1
        else:
            return [nums[l], nums[r]]

    return [nums[l], nums[r]]