LeetCode 1. 两数之和【二分查找、哈希、双指针】

难度： 简单

题目描述

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那两个整数，并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。
示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

思路1 一次遍历哈希

在进行迭代并将元素插入到表中的同时，还会检查表中是否已经存在当前元素所对应的目标元素。如果它存在，那我们已经找到了对应解，并立即将其返回。一遍哈希表的方法巧妙遵守了“不能重复使用同一个元素”这个约束条件。

代码

class Solution {
public:
    vector<int> twoSum(vector<int> &nums, int target) {
        unordered_map<int, int> map;
        for (int i = 0; i < nums.size(); ++i) {
            int tmp = target - nums[i];
            if (map.count(tmp)) {
                return vector<int>{i, map[tmp]};
            }
            map[nums[i]] = i;
        }
        throw "No two sum solution";
    }
};

思路2 二分查找

将数组排序并遍历，对于每个数，用二分查找确定是否有另外一个数满足要求。由于已经排好序了，所以在这个数后面的区域进行二分查找。

代码

class Solution {
public:
    vector<int> twoSum(vector<int> &nums, int target) {
        vector<pair<int, int>> pnums;
        for (int i = 0; i < nums.size(); ++i) {
            pnums.push_back(make_pair(i, nums[i]));
        }
        sort(pnums.begin(), pnums.end(), 
        [](pair<int, int>& a, pair<int, int>& b) { return a.second < b.second; });
        for (auto iter_a = pnums.begin(); iter_a + 1 != pnums.end(); ++iter_a) {
            int a = iter_a -> second;
            int b = target - a;
            auto iter_b = lower_bound(iter_a + 1, pnums.end(), b, [](const pair<int, int>& a,const int& b){ return a.second < b; });
            if (iter_b != pnums.end() && iter_b -> second == b) {
                return vector<int>{iter_a -> first, iter_b -> first};
            }
        }
        throw "No two sum solution";
    }
};

思路3 双指针

排序后用两个指针分别指向最前面和最后面，他们的和大于目标，说明要将后面的指针向前移，否则需要将前面的指针向后移。

代码

class Solution {
public:
    vector<int> twoSum(vector<int> &nums, int target) {
        vector<pair<int, int>> pnums;
        for (int i = 0; i < nums.size(); ++i)
            pnums.push_back(make_pair(i, nums[i]));
        
        sort(pnums.begin(), pnums.end(), 
        [](pair<int, int>& a, pair<int, int>& b) { return a.second < b.second; });
        
        int i = 0, j = pnums.size() - 1;
        while (pnums[i].second + pnums[j].second != target) {
            if (pnums[i].second + pnums[j].second > target) --j;
            else ++i;
        }
        return {pnums[i].first, pnums[j].first};
    }
};