哈希表复习 1.经典两数之和,分析为什么用unordered_map 因为map可以存储key和value,而set的key就是value 插入map的时候可以采用pair的方式,也可以用数组的形式
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 {};
}
};
四数之和:
class Solution {
public:
int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
unordered_map<int, int> umap; //key:a+b的数值,value:a+b数值出现的次数
// 遍历大A和大B数组,统计两个数组元素之和,和出现的次数,放到map中
for (int a : A) {
for (int b : B) {
umap[a + b]++;
}
}
int count = 0; // 统计a+b+c+d = 0 出现的次数
// 再遍历大C和大D数组,找到如果 0-(c+d) 在map中出现过的话,就把map中key对应的value也就是出现次数统计出来。
for (int c : C) {
for (int d : D) {
if (umap.find(0 - (c + d)) != umap.end()) {
count += umap[0 - (c + d)];
}
}
}
return count;
}
};
使用set的场景 当我们遇到了要快速判断一个元素是否出现集合里的时候,就要考虑哈希法了。
所以这道题目使用哈希法,来判断这个sum是否重复出现,如果重复了就是return false, 否则一直找到sum为1为止。注意,
class Solution {
public:
// 取数值各个位上的单数之和
int getSum(int n) {
int sum = 0;
while (n) {
sum += (n % 10) * (n % 10);
n /= 10;
}
return sum;
}
bool isHappy(int n) {
unordered_set<int> set;
while(1) {
int sum = getSum(n);
if (sum == 1) {
return true;
}
// 如果这个sum曾经出现过,说明已经陷入了无限循环了,立刻return false
if (set.find(sum) != set.end()) {
return false;
} else {
set.insert(sum);
}
n = sum;
}
}
};
总结:
-
需要「有序性 / 范围查找」→ 选红黑树系列(set/map/multiset/multimap)
- 仅存唯一值,要有序 →
set - 存键值对,要有序 →
map - 存重复值,要有序 →
multiset - 存重复键的键值对,要有序 →
multimap
- 仅存唯一值,要有序 →
-
追求「极致查找效率」,无需有序 → 选哈希表系列(unordered_set/unordered_map)
- 仅存唯一值,要高效去重 / 判重 →
unordered_set - 存键值对,要高效查值 →
unordered_map
- 仅存唯一值,要高效去重 / 判重 →
-
补充:自定义类型存储
-
红黑树系列:需提供比较函数(如
<运算符) -
哈希表系列:需提供哈希函数 + 相等判断函数(如
==运算符)
-