【C++】map & set
map & set
- 一、关联式容器
- 二、键值对
- 三、树形结构的关联式容器
-
- 1. set
-
- （1）set 的介绍
- （2）set 的使用
- 2. multiset
- 3. map
-
- （1）map 的介绍
- （2）map 的使用
- 4. multimap
- 四、map 和 set 的练习
-
- 1. 前K个高频单词
- 2. 两个数组的交集
一、关联式容器
我们在前面已经接触过 STL 中的部分容器，比如：vector、list、deque、等，这些容器统称为序列式容器(一级容器)，因为其底层为线性序列的数据结构，里面存储的是元素本身。
那什么是关联式容器？它与序列式容器有什么区别？关联式容器也是用来存储数据的，与序列式容器不同的是，其里面存储的是
二、键值对
用来表示具有一一对应关系的一种结构，该结构中一般只包含两个成员变量 key 和 value，key 代表键值，value 表示与 key 对应的信息。比如：现在要建立一个英汉互译的字典，那该字典中必然有英文单词与其对应的中文含义，而且，英文单词与其中文含义是一一对应的关系，即通过该应该单词，在词典中就可以找到与其对应的中文含义。
SGI-STL中关于键值对的定义：
template
struct pair
{
typedef T1 first_type;
typedef T2 second_type;
T1 first;
T2 second;
pair(): first(T1()), second(T2())
{}
pair(const T1& a, const T2& b): first(a), second(b)
{}
};
三、树形结构的关联式容器
根据应用场景的不同，STL 总共实现了两种不同结构的管理式容器：树型结构与哈希结构。树型结构的关联式容器主要有四种：map、set、multimap、multiset；这四种容器的共同点是：使用平衡搜索树(即红黑树) 作为其底层结果，容器中的元素是一个有序的序列。下面一依次介绍每一个容器。
1. set
（1）set 的介绍
首先我们可以看一下 set 的文档介绍：set 文档介绍.
简单概括：
-
set 是按照一定次序存储元素的容器
-
在 set 中，元素的 value 也标识它(value就是 key，类型为 T)，并且每个 value 必须是唯一的。
set 中的元素不能在容器中修改(元素总是const)，但是可以从容器中插入或删除它们。 -
set 在底层是用二叉搜索树(红黑树)实现的。
注意：
- 与 map/multimap 不同，map/multimap 中存储的是真正的键值对
- set 中插入元素时，只需要插入 value 即可，不需要构造键值对。
- set 中的元素不可以重复 (因此可以使用set进行去重) ；
- 使用 set 的迭代器遍历 set 中的元素，可以得到有序序列；
- set 中的元素默认按照小于来比较；
- set 中查找某个元素，时间复杂度为：O(logN).
（2）set 的使用
在使用之前我们先看一下 set 的模板参数列表：
其中：
- T: set 中存放元素的类型，实际在底层存储
- Compare：仿函数，set 中元素默认按照小于来比较
- Alloc：set 中元素空间的管理方式，使用STL提供的空间配置器管理
- insert
下面我们简单插入几个元素并打印出来：
void test1()
{
set s;
s.insert(2);
s.insert(6);
s.insert(1);
s.insert(4);
s.insert(9);
s.insert(9);
s.insert(9);
s.insert(0);
for (auto& e : s)
{
cout << e << " ";
}
cout << endl;
}
由于 set 底层有迭代器，所以可以使用范围 for. 运行结果如下：
我们看到 set 打印出来是有序的并且去重了；在这里我们可以看一下 insert 的返回值：
如上图，当我们插入的是一个值的时候，它的返回值是一个 pair 类型的键值对，所以我们可以简单用一个 pair 类型的变量接收，观察它对应的数据，如下：
void test1()
{
set s;
s.insert(2);
s.insert(6);
s.insert(1);
s.insert(4);
s.insert(9);
s.insert(0);
pair::iterator, bool> it1 = s.insert(9);
cout << *(it1.first) << " ";
cout << it1.second<< endl;
pair::iterator, bool> it2 = s.insert(10);
cout << *(it2.first) << " ";
cout << it2.second << endl;
}
如上代码，我们在已经有 9 的 s 中再次插入 9，我们观察它的返回值中的 first 和 second；然后我们插入 s 中没有的 10，继续观察；结果如下：
如上图，我们看到插入已有元素后，因为返回的 first 是个迭代器，所以需要解引用才能得到里面的值，里面的值给我们返回了已经存在的 9；而 second 返回了 0 即 false，代表插入失败。
插入没有的元素的时候 first 返回的是新插入的值所在节点的迭代器；second 则是 1，即 true，代表插入成功。
- erase
删除和我们往常用的差不多，直接删除需要删除的元素即可；注意删除成功返回 1；删除失败返回 0；
也可以直接给迭代器的位置直接删除，没有返回值。
如下代码：
void test1()
{
set s;
s.insert(2);
s.insert(6);
s.insert(1);
s.insert(4);
s.insert(9);
s.insert(0);
// 打印删除前
for (auto& e : s)
{
cout << e << " ";
}
cout << endl;
// 打印删除结果
cout << s.erase(1) << endl;
cout << s.erase(20) << endl;
// 打印删除后
for (auto& e : s)
{
cout << e << " ";
}
cout << endl;
}
结果如下：
- find
如上图我们看到，find 的返回值是 iterator，那么返回的是哪个位置的迭代器呢？我们继续看它的返回值：
如上，如果找到这个元素，则返回这个元素所以在位置的迭代器；否则返回 end() 位置。可以如下使用：
void test2()
{
set s;
s.insert(2);
s.insert(6);
s.insert(1);
s.insert(4);
s.insert(9);
s.insert(0);
set::iterator it = s.find(6);
if (it != s.end())
{
s.erase(it);
}
for (auto& e : s)
{
cout << e << " ";
}
cout << endl;
}
- count
count 返回值就是统计这个元素出现多少次，但是 set 的一个特性是去重，所以返回值只有 1 和 0；所以一般用来判断在不在；如下：
void test2()
{
set s;
s.insert(2);
s.insert(6);
s.insert(1);
s.insert(4);
s.insert(9);
s.insert(0);
if (s.count(6))
{
cout << "6在" << endl;
}
else
{
cout << "6不在" << endl;
}
if (s.count(10))
{
cout << "10在" << endl;
}
else
{
cout << "10不在" << endl;
}
}
- lower_bound 和 upper_bound
- lower_bound 是返回一个 >= val 值位置的 iterator ；
- upper_bound 是返回一个 > val 值位置的 iterator ；
使用如下：
void test3()
{
set::iterator itlow, itup;
set s;
// 插入 10 20 30 40 50 60 70 80 90
for (int i = 1; i < 10; i++)
s.insert(i * 10);
itlow = s.lower_bound(25);
itup = s.upper_bound(70);
s.erase(itlow, itup); // 删除区间 [30, 40, 50, 60, 70]
for (auto& e : s)
{
cout << e << " ";
}
cout << endl;
}
结果如下：
- equal_range
如上图介绍，equal_range 是返回一个 pair 类型，是一个范围的边界，该范围包括 s 中与 val 等价的所有元素。使用如下：
void test4()
{
set s;
// 插入 10 20 30 40 50 60 70 80 90
for (int i = 1; i < 10; i++)
s.insert(i * 10);
auto ret = s.equal_range(30);
cout << *ret.first << endl; // >= val
cout << *ret.second << endl; // > val
}
结果如下：
2. multiset
multiset 文档介绍
multiset 是按照特定顺序存储元素的容器，其中元素是可以重复的；它与 set 的区别就是 multiset 可以插入重复的元素。
- multiset 的使用
multiset 的许多接口都与 set 重复，所以它们的用法大体一致；
在这就介绍一下 find，如果有多个 val ，find 返回中序的第一个 val.
如下段代码：
void test5()
{
multiset ms;
ms.insert(2);
ms.insert(2);
ms.insert(1);
ms.insert(4);
ms.insert(2);
ms.insert(2);
ms.insert(0);
for (auto& e : ms)
cout << e << " ";
cout << endl;
multiset::iterator it = ms.find(2);
while (it != ms.end())
{
cout << *it << " ";
++it;
}
cout << endl;
}
结果如下，说明 find 是返回中序的第一个2：
3. map
（1）map 的介绍
我们先看一下 map 的文档介绍：map 文档介绍 .
简单概括：
- map 是关联容器，它按照特定的次序(按照 key 来比较)存储由键值 key 和值 value 组合而成的元素。
- 在 map 中，键值 key 通常用于排序和唯一地标识元素，而值value 中存储与此键值 key 关联的内容。键值 key 和值 value 的类型可能不同，并且在 map 的内部，key 与 value 通过成员类型 value_type 绑定在一起，为其取别名称为 pair：
typedef pair; 如下图，是文档内容的截取：value_type
- 在内部，map 中的元素总是按照键值 key 进行比较排序的。
- map 中通过键值访问单个元素的速度通常比 unordered_map 容器慢，但 map 允许根据顺序对元素进行直接迭代(即对map中的元素进行迭代时，可以得到一个有序的序列)。
- map 支持下标访问符，即在
[]中放入 key，就可以找到与 key 对应的 value。 - map 通常被实现为二叉搜索树(更准确的说：平衡二叉搜索树(红黑树))。
（2）map 的使用
- insert
因为 map 插入的是一对键值对，所以需要以 pair 的形式插入；我们先看一下 pair 的文档介绍：pair 文档介绍
其中它的构造函数如下：
但是我们习惯用以下这个接口初始化一个键值对对象：make_pair，如下：
所以我们插入的时候可以以如下两种方式插入：
void test6()
{
map dict;
dict.insert(pair("insert", "插入"));
dict.insert(make_pair("string", "字符串"));
map::iterator it = dict.begin();
while (it != dict.end())
{
cout << (*it).first << ":" << (*it).second << endl;
//cout << it->first << ":" << it->second << endl;
++it;
}
}
因为 iterator 迭代器封装的是一个个节点，所以解引用得到的是一个 pair 类型的键值对，所以解引用后使用 . 指定访问 first 还是 second；而 -> 我们在 list 部分讲过，当节点中存的类型是自定义类型的时候，我们使用 -> 就可以方便一点进行指定元素的访问，而 pair 正好就是自定义类型，所以用 -> 访问更方便。
注意，这里的是省略了一个 ->，实际上 it->first 就是 it.operator->()->first；其中 it.operator->() 是取到 pair 的地址，再使用 -> 即可进行指定元素访问。
然后其它接口的使用和 set 的用法差不多，只是 map 的类型变成了 pair 而已；所以其它接口不再进行介绍，下面开始介绍 map 中的最重要的接口：operator[].
- operator[]
如上图和下图，operator[] 的返回值是 pair 的第二个模板参数：
我们先看一下 operator[] 的使用，如下段代码：
void test7()
{
string array[] = { "苹果", "西瓜", "苹果", "西瓜", "苹果", "苹果", "西瓜", "苹果", "香蕉", "苹果", "香蕉" };
map mapCount;
for (auto& str : array)
mapCount[str]++;
for (auto& kv : mapCount)
cout << kv.first << ":" << kv.second << endl;
}
结果如下：
实际上，调用 operator[] 就是在调用：(*((this->insert(make_pair(k,mapped_type()))).first)).second
如下图是原文档的解释：
分析如下：
其中 mapped_type 是 map 模板中的第二个模板参数类型，mapped_type() 则是默认构造的匿名对象，所以我们传 int 类型的时候，如果 key 在 map 中没有，则初始化为 0；如果有则返回 key 对应的 value；所以当我们 ++ 的时候，就可以统计次数，其实 ++ 就是作用到 pair 的 value 上。
总结：
- map 中的的元素是键值对
- map 中的 key 是唯一的，并且不能修改
- 默认按照小于的方式对 key 进行比较
- map 中的元素如果用迭代器去遍历，可以得到一个有序的序列
- map 的底层为平衡搜索树(红黑树)，查找效率比较高 O(logN).
- 支持
[]操作符，operator[]中实际进行插入查找；
4. multimap
我们可以看一下文档介绍：multimap 文档介绍
multimap 和 map 的唯一不同就是：map 中的 key 是唯一的，而multimap 中 key 是可以重复的。
multimap 的使用可以参考 map，功能都是类似的，这里不再作介绍；
注意：
- multimap 中的 key 是可以重复的。
- multimap 中的元素默认将 key 按照小于来比较
- multimap 中没有重载 operator[] 操作，因为 key 是可以重复的，如果此时有多个 key，就不知道返回哪个 key 对应的 value了；而且当有 key 的时候，再来一个 key，就不知道是查找还是修改了。
- 使用时与map包含的头文件相同：
四、map 和 set 的练习
1. 前K个高频单词
题目链接 -> Leetcode -692.前K个高频单词
Leetcode -692.前K个高频单词
题目：给定一个单词列表 words 和一个整数 k ，返回前 k 个出现次数最多的单词。
返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率， 按字典顺序 排序。
示例 1：
输入 : words = [“i”, “love”, “leetcode”, “i”, “love”, “coding”], k = 2
输出 : [“i”, “love”]
解析 : “i” 和 “love” 为出现次数最多的两个单词，均为2次。
注意，按字母顺序 “i” 在 “love” 之前。
示例 2：
输入 : [“the”, “day”, “is”, “sunny”, “the”, “the”, “the”, “sunny”, “is”, “is”] , k = 4
输出 : [“the”, “is”, “sunny”, “day”]
解析 : “the”, “is”, “sunny” 和 “day” 是出现次数最多的四个单词，
出现次数依次为 4, 3, 2 和 1 次。
注意：
1 <= words.length <= 500
1 <= words[i] <= 10
words[i] 由小写英文字母组成。
k 的取值范围是[1, 不同 words[i] 的数量]
思路是先将 words 中的内容放入 map 中统计次数，再放入优先级队列中按照 cmp 方式进行比较确认优先级；其中 cmp 方法是次数多的优先，如果次数多的就按照字典顺序比较；代码如下：
class Solution
{
public:
// 仿函数，按照 pair 中的 second 比较，降序；如果 second 相等，按照字典顺序排序
class cmp
{
public:
bool operator()(const pair& p1, const pair& p2)
{
return p1.second < p2.second || (p1.second == p2.second && p1.first > p2.first);
}
};
vector topKFrequent(vector& words, int k)
{
priority_queue, vector>, cmp> pq;
map mp;
// 先放入 map 中统计次数
for (auto& str : words)
mp[str]++;
// 再放入优先级队列中
for (auto& str : mp)
pq.push(str);
// 取优先级队列的前 k 个
vector ret;
while (k--)
{
ret.push_back(pq.top().first);
pq.pop();
}
return ret;
}
};
2. 两个数组的交集
题目链接 -> Leetcode -349.两个数组的交集
Leetcode -349.两个数组的交集
题目：给定两个数组 nums1 和 nums2 ，返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
示例 1：
输入：nums1 = [1, 2, 2, 1], nums2 = [2, 2]
输出：[2]
示例 2：
输入：nums1 = [4, 9, 5], nums2 = [9, 4, 9, 8, 4]
输出：[9, 4]
解释：[4, 9] 也是可通过的
提示：
1 <= nums1.length, nums2.length <= 1000
0 <= nums1[i], nums2[i] <= 1000
思路是将两个数组分别放入 set 中，然后使用双指针找交集； 具体思路如下代码：
class Solution {
public:
vector intersection(vector& nums1, vector& nums2)
{
// 将 nums1 和 nums2 分别放入 set 中
set s1(nums1.begin(), nums1.end());
set s2(nums2.begin(), nums2.end());
// 使用双指针的思路
// 两个值相等则是交集，同时++；
// 两个值不相等则小的那个++；
// 直到其中一个结束
vector ret;
set::iterator it1 = s1.begin(), it2 = s2.begin();
while (it1 != s1.end() && it2 != s2.end())
{
if (*it1 == *it2)
{
ret.push_back(*it1);
++it1, ++it2;
}
else if (*it1 < *it2)
{
++it1;
}
else
{
++it2;
}
}
return ret;
}
};