力扣刷题 DAY_16 哈希表

leetcode 1

链接： 力扣 。

题目：

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那两个整数，并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

示例：

输入：nums = [2, 7, 11, 15], target = 9

输出：[0, 1]

解释：nums[0] + nums[1] = 2 + 7 = 9

思路：

最直接想到的方法是暴力双循环，时间复杂度为O(n^2)。

更优的方法是使用hash表来做。对于本题应该选择的容器为map，因为我们需要成对的值。而STL中map有三种：map、multimap、unordered_map。map和multimap底层是红黑树，会对元素按键值排序。我们并不需要排序，故选择unordered_map。同时，unordered_map底层是哈希表，查找效率高于红黑树。

先建立一个unoredered_map容器，元素的值为key，元素的下标为value，这样做是为了方便查找元素。然后遍历数组nums，对于遇到的每个元素nums[i]，查找在哈希表中是否有target-nums[i]，如果存在，说明我们找到满足要求的两个数，输出当前的nums[i]和哈希表中查找到的那个数；如果不存在，就将nums[i]加入哈希表，继续向后遍历。

参考代码：

class Solution {
public:
    vector twoSum(vector& nums, int target) {
        unordered_map hashMap;
        for (int i = 0; i < nums.size(); i++) {
            auto it = hashMap.find(target - nums[i]);
            if (it == hashMap.end()) {
                hashMap.insert(pair(nums[i], i));
            }
            else {
                return {it->second, i};
            }
        }
        return {};
    }
};