本题使用了哈希表(unordered map)的数据结构,关键点在于掌握map容器的用法。map的find函数是按照索引进行查找并不是值查找,所以该题解中map的第一个元素设为nums数组中的值,第二个元素设为对应在nums数组中的位置。
时间复杂度:O(n),空间复杂度:O(n)
思路:
- 定义一个 unordered_map,用来记录已经遍历过的数字以及其下标。
- 遍历给定的 nums 数组,对于每个元素 nums[i],执行以下步骤:
- 在 map 中查找是否存在与 target-nums[i] 相等的数字,如果找到了,就返回这两个数的下标(通过 map 的下标运算符实现)。
- 如果没找到匹配的数字,将当前元素 nums[i] 和其下标 i 加入到 map 中。
- 如果 nums 中不存在这样的两个数,则返回一个空的 vector。
代码:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
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};
} else {
// 如果没找到匹配对,就把访问过的元素和下标加入到map中
map.insert(pair<int, int>(nums[i], i));
}
}
return {};
}
};