力扣链接:https://leetcode.cn/problems/two-sum/submissions/
题目描述:
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
解题思路:
遍历数组,如nums[i] = 2
这时想到去数组中查找target-nums[i]
即7是否存在,如果存在返回7所在的下标位置;这时想到C++中哪种数据结构可以同时记录值-下标
信息?当然就是map这种数据结构。
在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) |
本题使用无序列表即可,map中查找函数find(), count()
- count()函数 元素存在返回1,不存在返回0
- find()函数 返回找到元素的下标
解题代码:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
std::unordered_map<int, int> myMap;
for (int i = 0; i < nums.size(); i++) {
int isExist = myMap.count(target-nums[i]);
if (isExist == 1) {
return vector<int> {myMap[target-nums[i]], i};
} else {
myMap[nums[i]] = i;
}
}
return vector<int> {-1, -1};
}
};
参考代码2:来源代码随想录【https://programmercarl.com/0001.%E4%B8%A4%E6%95%B0%E4%B9%8B%E5%92%8C.html#%E6%80%9D%E8%B7%AF】
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
std::unordered_map <int,int> map;
for(int i = 0; i < nums.size(); i++) {
// 遍历当前元素,并在map中寻找是否有匹配的key
auto iter = map.find(target - nums[i]);
if(iter != map.end()) {
return {iter->second, i};
}
// 如果没找到匹配对,就把访问过的元素和下标加入到map中
map.insert(pair<int, int>(nums[i], i));
}
return {};
}
};