题目描述:给定数组与目标值,返回数组nums中两数和为目标值target的两数对应的数组下标
题目链接:https://leetcode-cn.com/problems/two-sum
- 暴力解法
暴力解法直接分别让整个数组中所有数之间互相求和与目标值进行对比,当满足条件时返回两数对应的数组下标,代码如下:
// 暴力解法
class Soultion
{
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 {};
}
};
- 哈希表解法
使用哈希表可以极大程度的降低程序的时间复杂度,哈希表的建立使用C++中的unordered_map容器,其内部使用哈希表实现,此题对排序没有要求,执行速度上要高于map容器。
哈希表的key值为数组中的数据nums[i],value值为数据对应的数组下标i。
算法为依次检测哈希表中是否存在key值与 target-nums[i] 相等,相等则找到答案,不相等则将新值加入到哈希表中。代码如下:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> m;
auto it = m.end();
for (int i = 0; i < nums.size(); i++) {
//检测m中是否存在对应的键值
if ((it = m.find(target - nums[i])) != m.end())
return { (*it).second, i };
// m[target-nums[i]]为已经加入map的元素的索引,所以小于本轮循环中的i,放在前面
//向map中添加元素
m.insert(make_pair(nums[i], i));
}
return {};
}
};
总结
unordered_map容器的使用,通过其find方法判断key值是否存在,以及返回的迭代器中使用first和second中分别获取key值和value值,通过insert方法以及make_pair将对组插入到map/unordered_map中。