剑指Offer57-和为s的两个数字

剑指 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 <= nums.length <= 10^5

  • 1 <= nums[i] <= 10^6

来源:力扣(LeetCode)

链接:力扣

著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

一解

使用双指针 l, r 指向数组的左和右:

  1. 计算和 s = nums[i] + nums[j]

  2. s > targets,则指针 j 向左移动,即执行 j = j - 1

  3. s < targets,则指针 i 向右移动,即执行 i = i + 1

  4. s = targets,立即返回数组 [nums[i], nums[j]]


class Solution:

    def twoSum(self, nums: List[int], target: int) -> List[int]:

        l, r = 0, len(nums) - 1

        while l < r:

            s = nums[l] + nums[r]

            if s == target:

                return nums[l], nums[r]

            elif s < target:

                l += 1

            else: r -= 1

        

  • 时间复杂度:O(N)

  • 空间复杂度:O(1)