哈希表 映射
一、题源:
1. 两数之和
问题描述
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
二、 思路:
哈希表
关于哈希表一般使用到的数据结构有 数组 、 set(集合) 、 map(映射) 。
之前俩篇文章分别介绍了数组和集合的相关题目,使用数组一般是针对数值大小有限且机制的情况。 set 只能存储key,而这道题既需要存储数值,也需要存储下标,所以需要用到第三种数据结构 map。
对应 set,C++ 的关于 map 的实现都有三种,分别是:
映射 | 底层实现 | 是否有序 | 数值是否可以重复 | 能否更改数值 | 查询效率 | 增删效率 |
---|---|---|---|---|---|---|
std::map | 红黑树 | key有序 | key不可重复 | key不可修改 | O ( l o g n ) O(log n) O(logn) | O ( l o g n ) O(log n) O(logn) |
std::multimap | 红黑树 | key有序 | key可重复 | key不可修改 | O ( l o g n ) O(log n) O(logn) | O ( l o g n ) O(log n) O(logn) |
std::unordered_map | 哈希表 | key无序 | key不可重复 | key不可修改 | O ( 1 ) O(1) O(1) | O ( 1 ) O(1) O(1) |
这道题并不需要对数据进行排序,而且也不需要让数据重复,所以选择查询和增删效率最好的 unordered_map 。
参考代码如下
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> map;
vector<int> result;
for(auto it = nums.begin(); it != nums.end(); it++)
{
if(map.find(*it) == map.end()) map[target - *it] = it - nums.begin();
else
{
result.push_back(it - nums.begin());
result.push_back(map[*it]);
return result;
}
}
return result;
}
};
时间复杂度:
O
(
n
)
O(n)
O(n)
空间复杂度:
O
(
1
)
O(1)
O(1)
三、相关题目及C++实现
49. 字母异位词分组
问题描述:
给你一个字符串数组,请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。
字母异位词 是由重新排列源单词的字母得到的一个新单词,所有源单词中的字母通常恰好只用一次。
实现:
class Solution {
public:
vector<vector<string>> groupAnagrams(vector<string>& strs) {
vector<vector<string>> result;
unordered_map<string, vector<string>> map;
for(auto it = strs.begin(); it != strs.end(); ++it)
{
string key = *it;
sort(key.begin(), key.end());
map[key].push_back(*it);
}
for (auto it = map.begin(); it != map.end(); ++it) result.push_back(it->second);
return result;
}
};
时间复杂度:
O
(
n
)
O(n)
O(n)
空间复杂度:
O
(
1
)
O(1)
O(1)
350. 两个数组的交集 II
问题描述:
给你两个整数数组 nums1 和 nums2 ,请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数,应与元素在两个数组中都出现的次数一致(如果出现次数不一致,则考虑取较小值)。可以不考虑输出结果的顺序。
实现:
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
unordered_map<int,int> map;
vector<int> result;
for(auto& num : nums1) ++map[num];
for(auto& num : nums2) if( --map[num] >= 0) result.push_back(num);
return result;
}
};
时间复杂度:
O
(
n
)
O(n)
O(n)
空间复杂度:
O
(
1
)
O(1)
O(1)
454. 四数相加 II
问题描述:
给你四个整数数组 nums1、nums2、nums3 和 nums4 ,数组长度都是 n ,请你计算有多少个元组 (i, j, k, l) 能满足:
- 0 <= i, j, k, l < n
- nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
实现:
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int,int> map;
int count = 0;
for(auto it1 = nums1.begin(); it1 != nums1.end(); ++it1)
for(auto it2 = nums2.begin(); it2 != nums2.end(); ++it2) ++map[-*it1 - *it2];
for(auto it1 = nums3.begin(); it1 != nums3.end(); ++it1)
for(auto it2 = nums4.begin(); it2 != nums4.end(); ++it2) count += map[*it1 + *it2];
return count;
}
};
时间复杂度:
O
(
n
2
)
O(n^2)
O(n2)
空间复杂度:
O
(
1
)
O(1)
O(1)