剑指 Offer 57. 和为s的两个数字 哈希 双指针 二分查找

地址 https://leetcode-cn.com/problems/he-wei-sde-liang-ge-shu-zi-lcof/

输入一个递增排序的数组和一个数字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

解答

递增有序的数组 寻找两个数 ，这样的条件就宽泛了很多。

有序的数组可以考虑使用 二分查找，复杂度为O(nlogⁿ)

还可以和leetcode 第一题两数之和解法类似，可以使用双指针和哈希的做法 复杂度O(n);

 

1 二分查找

class Solution {
public:
    bool Check(const vector<int>& v, int idx, int find) {
        return v[idx] >= find;
    }

    bool bSearch(const vector<int>& v, int l, int r, int find)
    {
        bool ret = false;
        if (l < 0 || l >= v.size() || r < 0 || r >= v.size()) return ret;

        while (l < r) {
            int mid = l+r >> 1;
            if (Check(v,mid,find)) r = mid;
            else l = mid + 1;
        }

        return v[l] == find;
    }

    vector<int> twoSum(vector<int>& nums, int target) {
        for (int i = 0; i < nums.size(); i++) {
            int find = target - nums[i];
            bool ret = false;
            if (find > nums[i]) {
                ret = bSearch(nums,i+1,nums.size()-1,find);
                if (ret) {
                    return vector<int>{nums[i], find};
                }
            }
            else if (find < nums[i]) {
                ret = bSearch(nums, 0, i - 1, find);
                if (ret) {
                    return vector<int>{nums[i], find};
                }
            }
            else {
                //说明查找的数字和nums[i]相等 直接查找该数字前后是否有该数即可
                if (i - 1 >= 0 && nums[i - 1] == nums[i]) return vector<int>{find, find};
                if (i + 1 < nums.size() && nums[i + 1] == nums[i]) return vector<int>{find, find};
            }
        }

        return vector<int>();
    }
};

 

 

2 哈希

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

        return vector<int>();
    }
};

 

 

 

3 双指针

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