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

题目描述

和为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

算法

此题与Leetcode1两数之和区别在与本题多了一个条件：输入数组是递增的。
由于题目给定的数组的长度是$10^5$，所有时间复杂度必须限制到$O(nlog(n))$。

解法一

同Leetcode1的解法，遍历数组判断哈希表中是否存在s减去当前数的数字，如果存在则返回当前下数的下标和对应的数的下标。
此方法的时间复杂度为$O(n)$，空间复杂度为$O(n)$

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_set<int> hash;
        for(int i = 0; i < nums.size(); i++) {
            if(hash.count(target - nums[i])) return {nums[i], target - nums[i]};
            hash.insert(nums[i]);
        }
        return {0,0};
    }
};

解法二

由于数组是递增的，那个当两个数nums[i] < nums[j]，如果他们的和比s小时，必然是i往后移动一位；如果他们的和比s大时，必然是j往前移动一位。
因此可以用两个变量i和j标识下标，然后根据nums[i]与nums[j]的和与s的大小关系移动i和j，直到找到一组满足要求数或者i与j相遇。
时间复杂度$O(n)$，空间复杂度$O(1)$

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        for(int i = 0, j = nums.size() - 1; i < j; ) {
            if(nums[i] + nums[j] > target) j--;
            else if(nums[i] + nums[j] < target) i++;
            else return {nums[i], nums[j]};
        }
        return {};
    }
};