题目
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
输出示例
解法一:暴力法,时间复杂度O( n 2 n^2 n2)
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
for (int i = 0; i < nums.size(); i++) {
for (int j = i + 1; j < nums.size(); j++) {
if (nums[i] + nums[j] == target) {
return { i,j };
}
}
}
return {};
}
};
解法二:哈希表,时间复杂度O(n),学习答案!!
思路及算法
注意到解法一的时间复杂度较高的原因是寻找
target - x
的时间复杂度过高。因此,我们需要一种更优秀的方法,能够快速寻找数组中是否存在目标元素。如果存在,我们需要找出它的索引。
使用哈希表,可以将寻找
target - x
的时间复杂度降低到从O(N)
降低到O(1)
。 创建一个哈希表,对于每一个x
,首先查询哈希表中是否存在target - x
,然后将x
插入到哈希表中,即可保证不会让x
和自己匹配。
class Solution {
public:
//map的key是nums[i]的值,value是下标i
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> hashtable;
for (int i = 0; i < nums.size(); ++i) {
//auto被定义为自动推断变量的类型
auto it = hashtable.find(target - nums[i]);
//匹配成功
if (it != hashtable.end()) {
return {it->second, i};
}
//匹配不成功存入哈希表
hashtable[nums[i]] = i; //下标i是hash值,nums[i]是key值
}
return {};
}
};
作者:力扣官方题解
链接:https://leetcode.cn/problems/two-sum/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
哈希表使用:此文详细描述哈希表,请点此处查看
一、unordered_map容器
- 头文件:
#include<unordered_map>
- 创建表
unordered_map<key,value> Map_name;
- 插入元素
①a[key]=value;
②Map_name.insert(pair<int,int>(key,value));
③unordered_map<int,int>Map_name={{key1,value1},{key2,value2}};
- 查找键值
table_name.find(key)
以key作为参数寻找哈希表中的元素,如果哈希表中存在该key值则返回该位置上的迭代器,否则返回哈希表最后一个元素下一位置上的迭代器,当已经存在键值的情况返回end() - 创建迭代器
unordered_map<key,value>::iterator iterator_name;
- 用迭代器访问元素的键值
it->first
- 用迭代器访问元素的键值对应的元素值
it->second
end( )函数
:返回一个指向哈希表结尾位置的下一个元素的迭代器
二、C++中auto被定义为自动推断变量的类型:
auto it = m.find(target - nums[i]);
//这里it被自动推断为map<int, int>::iterator类型