【剑指Offer系列57-1】和为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

代码

Python

# 思路：
# 双指针法
# 复杂度：
# O(N)
class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        l,r=0,len(nums)-1 # 双指针初始化
        res=[] # 结果列表初始化
        while l<r:
            s=nums[l]+nums[r]
            if s==target:
                res.append(nums[l])
                res.append(nums[r])
                break
            elif s>target:
                r-=1
            else:
                l+=1
        return res

C++

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        int l=0,r=nums.size()-1;
        vector<int> res;
        while (l<r) {
            int s=nums[l]+nums[r];
            if (s==target) {
                res.push_back(nums[l]);
                res.push_back(nums[r]);
                break;
            }
            else if (s>target) {
                r--;
            }
            else
                l++;
        }
        return res;
    }
};