给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
如果没有符合条件的两个元素,就返回空数组,或者有多对符合条件的,只返回一对就可以。
再强调一下 什么时候使用哈希法,当我们需要查询一个元素是否出现过,或者一个元素是否在集合里的时候,就要第一时间想到哈希法。
本题呢,就需要一个集合来存放我们遍历过的元素,然后在遍历数组的时候去询问这个集合,某元素是否遍历过,也就是 是否出现在这个集合。
那么我们就应该想到使用哈希法了。
因为本题,我们不仅要知道元素有没有遍历过,还要知道这个元素对应的下标,需要使用 key value结构来存放,key来存元素,value来存下标,那么使用map正合适。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
std::unordered_map <int,int> map; //定义容器map
for(int i = 0; i < nums.size(); i++) {
// 遍历当前元素,并在map中寻找是否有匹配的key
auto iter = map.find(target - nums[i]);//find函数返回目标元素的迭代器,auto 是 C++11 引入的关键字,用于自动推导变量的类型。
if(iter != map.end()) {
return {iter->second, i};//iter是迭代器变量,是指针,iter->first是key值,second是value值。
}
// 如果没找到匹配对,就把访问过的元素和下标加入到map中
map.insert(pair<int, int>(nums[i], i)); //对于map这种键值对的数据存储结构,在插入新键值对时,需要用pair来组合一对数据插入。
}
return {};
}
};
注意:
1.auto的用法,auto可以自动推导接收的变量的数据类型
2.iter迭代器在map中,->first和second分别是指向key和value。
3.map插入新的键值对时,需要用pair来组合一对数据。
来自:代码随想录