题目
给定一个整数数组 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 <= nums.length <= 104
-
-109 <= nums[i] <= 109
-
-109 <= target <= 109
-
只会存在一个有效答案
解法
哈希表
基本思路:暴力题解的时间复杂度较高的原因是寻找 target - x 的时间复杂度过高。因此,我们需要一种能够快速寻找数组中是否存在目标元素并返回索引的方法。使用哈希表,可以将寻找 target - x 的时间复杂度降低到从 O(N) 降低到 O(1)。 创建一个哈希表,对于每一个 x,我们首先查询哈希表中是否存在 target - x,然后将 x 插入到哈希表中,即可保证不会让 x 和自己匹配。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> mp;
for( int i=0; i<nums.size(); i++ ){
if( mp.count(target-nums[i]) )
return vector<int>{i, mp[target-nums[i]]};
else mp[nums[i]] = i;
}
return vector<int>();
}
};
暴力题解
基本思路:枚举数组中的每一个数 x,在 x 后面的元素中寻找是否存在 target - x。
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(N2),其中 N是数组中的元素数量。最坏情况下数组中任意两个数都要被匹配一次。
所用到的数据结构
哈希表unordered_map
优点:内部实现了哈希表,因此其查找速度是常量级别的。
缺点:哈希表的建立比较耗费时间
适用处:对于查找问题,unordered_map会更加高效一些,因此遇到查找问题,常会考虑一下用unordered_map
定义
#include<unordered_map> unorderede_map<int, int> s;
函数
s.clear()
:清空
s.begin()
:指向map中第一个元素的迭代器
s.end()
:指向map中最后一个元素的下一位的迭代器,与vector类似,即前闭后开
s.erase(it)
:从map中删除迭代器it指向的元素,时间复杂度O(logn)
s.erase({key, value})
:从map中删除所有相等的元素,时间复杂度O(logn+k)
s.insert({key, value})
:插入map,如果已存在,则不会重复插入
s.find(x_key)
:查找key为x_key的二元组
s.count(x_key)
:如果容器具有与键 k 关联的值,则返回 1,否则返回 0
总结
在数组中寻找符合某个条件的数并返回标签,可以用哈希表,其中s[nums[i]] = i
可以存储索引,s[nums[i]] = k
可以存储其他有效信息。