每日一解 unordered-map解决两数之和

两数之和

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

给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum

解题思路

第一时间想到的解题思路肯定是暴力法。因为确实比较容易，就不展开说了，直接贴一段随手敲的暴力解法：

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

但是只会暴力解法很明显是不够的，所以根据提示的解题思路，采用哈希表对题目进行处理。但是哈希表具体使用的知识掌握不够全面，所以专门去了解了一下STL中unordered-map的具体使用。

unordered-map及解答

首先，本问题其实是一个查找问题。因此采用内部实现了哈希表的unorder-map比较适合，因为其查找速度非常的快，哈希表的查找速度是O(1)，当然最坏情况可能会达到O(n)。
使用unorder-map解决本问题的思路也很简单，将数组里每个数都放入unorder-map中，然后依次检查每个数和目标结果减去该数是否在unorder-map中即可。
那么直接放代码：

class Solution {
public:
	vector<int> twoSum(vector<int>& nums, int target) {
		unordered_map<int, int> data;
		vector<int> answer;
		for (int i = 0; i < nums.size(); i++) {
			data.insert(pair<int, int>(nums[i], i));
		}
		for (int i = 0; i < nums.size(); i++) {
			if(data.find(target - nums[i]) != data.end()){//注意点
				if (data.find(target - nums[i])->second == i) {
					continue;
				}
				answer.push_back(i);
				answer.push_back(data.find(target - nums[i])->second);
				return answer;
			}
		}
		return answer;
	}
};

这里在结尾的地方稍微注释两点，一个是unordered_map的find函数返回为迭代器，指向的first是存起来的key值，second则是存放的value值。另一个则是find函数找不到对应的key时，返回的会是unordered_map的end，因此对比一下返回的迭代器是否为end就是判断该值是否存在于map中的方法。

顺便一提，unordered_map的头文件为

#include <unordered_map>