剑指offer 57 - 和为S的两个数字 - python

95 阅读1分钟

题目描述:

输入一个递增排序的数组和一个数字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 < = n u m s . l e n g t h < = 1 0 5 1<= nums.length <= 10^5 1<=nums.length<=105
1 < = n u m s [ i ] < = 1 0 6 1 <= nums[i] <= 10^6 1<=nums[i]<=106


双指针法

设置两个指针leftright分别指向数组的头和尾

  • 如果 a r r a y [ l e f t ] + a r r a y [ r i g h t ] < S array[left] + array[right] < S array[left]+array[right]<S,说明当前元素和小了,则left右移
  • 如果 a r r a y [ l e f t ] + a r r a y [ r i g h t ] > S array[left] + array[right] > S array[left]+array[right]>S,说明当前元素和大了,则right左移
  • 如果 a r r a y [ l e f t ] + a r r a y [ r i g h t ] = = S array[left] + array[right] == S array[left]+array[right]==S,则直接返回 [ a r r a y [ l e f t ] , a r r a y [ r i g h t ] ] [array[left], array[right]] [array[left],array[right]]即可
class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        if not nums: return []

        left, right = 0, len(nums) - 1
        while left < right:
            if nums[left] + nums[right] == target:
                return [nums[left], nums[right]]
            elif nums[left] + nums[right] < target:
                left += 1
            else:
                right -= 1

        return []