1题目
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
输入:nums = [3,2,4], target = 6
输出:[1,2]
示例 3:
输入:nums = [3,3], target = 6
输出:[0,1]
2链接
视频:梦开始的地方,Leetcode:1.两数之和,学透哈希表,map使用有技巧!_哔哩哔哩_bilibili
3解题思路
当我们需要查询一个元素是否出现过,或者一个元素是否在集合里的时候,就要第一时间想到哈希法。本题需要一个集合来存放我们遍历过的元素,然后在遍历数组的时候去询问这个集合,某元素(target-nums[i])是否出现在这个集合。
对于本题,我们不仅要知道元素有没有遍历过,还有知道这个元素对应的下标,需要使用 key value(键值对)结构来存放,key来存元素,value来存下标。因为map.find()做查询操作时返回的是key,我们需要的是查询曾经nums存进来的元素,所以要用数组元素作为key,数组元素下标作为value来存储进map。
使用数组和set来做哈希法的局限。
数组的大小是受限制的,而且如果元素很少,而哈希值太大会造成内存空间的浪费。
set是一个集合,里面放的元素只能是一个key,而两数之和这道题目,不仅要判断y是否存在而且还要记录y的下标位置,因为要返回x 和 y的下标。所以set 也不能用。
此时就要选择另一种数据结构:map ,map是一种key value的存储结构,可以用key保存数值,用value在保存数值所在的下标。
C++中map,有三种类型:
映射 | 底层实现 | 是否有序 | 数值是否可以重复 | 能否更改数值 | 查询效率 | 增删效率 |
std::map | 红黑树 | key有序 | key不可重复 | key不可修改 | O(log n) | O(log n) |
std::multimap | 红黑树 | key有序 | key可重复 | key不可修改 | O(log n) | O(log n) |
std::unordered_map | 哈希表 | key无序 | key不可重复 | key不可修改 | O(1) | O(1) |
std::unordered_map 底层实现为哈希表,std::map 和std::multimap 的底层实现是红黑树。
std::map 和std::multimap 的key是有序的(这个问题也经常作为面试题,考察对语言容器底层的理解)。
这道题目中并不需要key有序,选择std::unordered_map 效率更高!
在遍历数组的时候,只需要向map去查询是否有和目前遍历元素比配的数值,如果有,就找到的匹配对,如果没有,就把目前遍历的元素放进map中,因为map存放的就是我们访问过的元素。
过程如下:
![](https://img-blog.csdnimg.cn/img_convert/7f2eaf9f6084556d0fa1343f746e5cb6.png)
![](https://img-blog.csdnimg.cn/img_convert/97e89f27ddbec1075e5b87fca7bb3a49.png)
4代码
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> map;
for (int i = 0; i < nums.size(); i++) {
//遍历到nums[i]时,在map中待查找的数
int findNum = target - nums[i];
// 遍历当前元素,并在map中寻找是否有匹配的key,这里的key对应nums中的元素,value对应nums中的下标
if (map.find(findNum) != map.end()) {
return {map.find(findNum)->second, i};
}
//引用传递一个键值对pair<iterator,bool> insert (const value_type& val);
map.insert(pair<int,int>(nums[i],i));
}
return {};
}
};
5补充知识
5.1map容器的迭代器first、second用法
例:
map<string, int> m_stlmap;
m_stlmap[“xiaomi”] = 88;
auto mpit = m_stlmap.begin();
first会得到Map中key的有效值,
second会得到Map中value的有效值。
所以
mpit ->first; // 得到是 string 值是 “xiaomi”
mpit ->second; //得到是 int 值是 88
5.2map取值的find、[]、at方法特性对比
find用法特点:
有就是有,没有就是没有,需要判断find的返回结果,才知道有没有
#include <map>
#include <iostream>
using namespace std;
int main()
{
map<string, int> m;
m["xiaoming"] = 80;
m["xiaogang"] = 90;
m["xiaoqing"] = 100;
auto iter = m.find("xiaoqiang");
if (iter != m.end())
{
// 找到了
cout << "xiaoqiang's score: " << iter->second << endl;
}
else
{
// not found
cout << "not find xiaoqiang's score" << endl; // 应该输出该行
}
return 0;
}
[]的特点:
1.使用简单不管有没有,都是有。因为没有就是0,字符串就是空。
2.对于示例中给map插入值的操作,也是一样,如果原先不存在该key,则插入,如果存在,则覆盖插入。
#include <map>
#include <iostream>
using namespace std;
int main()
{
try
{
map<string, int> m;
m["xiaoming"] = 80;
m["xiaogang"] = 90;
m["xiaoqing"] = 100;
cout << "xiaoqing's score: " << m.at("xiaoqing") << endl; // 输出100
cout << "xiaoqiang's score" << m.at("xiaoqiang") << endl; // 抛出异常:out_of_range exception
}
catch(std::exception& e)
{
cout << "catch exception: " << e.what() << endl;
}
return 0;
}
at的使用特点
相比于find,能少敲键盘;相比于[],更安全
如果存在,则返回它的值,如果不存在,则抛出异常
3.常用于待查找的key都应该存在的情况,如果查找的key不存在,则说明程序有问题了
#include <map>
#include <iostream>
using namespace std;
int main()
{
try
{
map<string, int> m;
m["xiaoming"] = 80;
m["xiaogang"] = 90;
m["xiaoqing"] = 100;
cout << "xiaoqing's score: " << m.at("xiaoqing") << endl; // 输出100
// 抛出异常:out_of_range exception
cout << "xiaoqiang's score" << m.at("xiaoqiang") << endl;
}
catch(std::exception& e)
{
cout << "catch exception: " << e.what() << endl;
}
return 0;
}
总结:
如果纯粹是查找取值,不要用[]操作符,因为它会自动创建不存在的元素
正因为上面的原因,[]仅用于非const的情形下。在const map下使用[]会导致编译期错误
如果不确定待查找的值是否存在,且不存在也是合理的,则使用find,判断返回值是否是end()
如果待查找的值肯定存在,如果不存在就是逻辑上的错误,则使用at,处理out_of_range异常