在std::map
数据结构中,键-值通常都对应存在,而且键通常是唯一并排序过的,而且键值一旦设定那么就不允许用户再进行修改。为了阻止用户修改键,键的类型声明使用了const
。
这种限制是非常明智的,其可以保证用户很难在使用std::map
的时候出错。不过,如果我们真的需要修改map
的键值该怎么办呢?
C++17之前,因为对应的键已经存在,我们不得不将整个键-值对从树中移除,然后再插入。这种方法的确定很明显,其需要分配出一些不必要的内存,感觉上也会对性能有一定的影响。
从C++17起,我们无需重新分配内存,就可以删除和重新插入map键值对。下面的内容中将会展示如何操作。
我们使用std::map
类型一个实现应用,用于确定车手在虚拟比赛中的排位。当车手在比赛中完成超越,那么我们将使用C++17的新方法改变其键值。
-
包含必要的头文件和声明使用的命名空间。
#include <iostream> #include <map> using namespace std;
-
我们会在修改map的时候打印之前和之后结果,所以这里先实现了一个辅助函数。
template <typename M> void print(const M &m) { cout << "Race placement:\n"; for (const auto &[placement, driver] : m) { cout << placement << ": " << driver << '\n'; } }
-
主函数中,我们实例化并初始化一个
map
,其键为整型,表示是当前的排位;值为字符型,表示驾驶员的姓名。我们在这里先打印一下这个map
,因为我们会在下一步对其进行修改。int main() { map<int, string> race_placement { {1, "Mario"}, {2, "Luigi"}, {3, "Bowser"}, {4, "Peach"}, {5, "Yoshi"}, {6, "Koopa"}, {7, "Toad"}, {8, "Donkey Kong Jr."} }; print(race_placement);
-
让我来看下排位赛的某一圈的情况,Bowser因为赛车事故排在最后,Donkey Kong Jr. 从最后一名超到第三位。例子中首先要从
map
中提取节点,因为这是唯一能修改键值的方法。extract
函数是C++17新加的特性。其可以从map
中删除元素,并没有内存重分配的副作用。看下这里是怎么用的吧。{ auto a(race_placement.extract(3)); auto b(race_placement.extract(8));
-
现在我们要交换Bowser和Donkey Kong Jr.的键。键通常都是无法修改的,不过我们可以通过
extract
方法来修改元素的键。swap(a.key(), b.key());
-
std::map
的insert
函数在C++17中有一个新的重载版本,其接受已经提取出来的节点,就是为了在插入他们时,不会分配不必要的内存。race_placement.insert(move(a)); race_placement.insert(move(b)); }
-
最后,我们打印一下目前的排位。
print(race_placement); }
-
编译并运行可以得到如下输出。我们可以看到初始的排位和最后的排位。
$ ./mapnode_key_modification Race placement: 1: Mario 2: Luigi 3: Bowser 4: Peach 5: Yoshi 6: Koopa 7: Toad 8: Donkey Kong Jr. Race placement: 1: Mario 2: Luigi 3: Donkey Kong Jr. 4: Peach 5: Yoshi 6: Koopa 7: Toad 8: Bowser
在C++17中,std::map
有一个新成员函数extract
。其有两种形式:
node_type extract(const_iterator position);
node_type extract(const key_type& x)
在例子中,我们使用了第二个,能接受一个键值,然后找到这个键值,并提取对应的map
节点。第一个函数接受一个迭代器,提取的速度会更快,应为给定了迭代器就不需要在查找。
当使用第二种方式去提取一个不存在的节点时,会返回一个空node_type
实例。empty()
成员函数会返回一个布尔值,用来表明node_type
实例是否为空。以任何方式访问一个空的实例都会产生未定义行为。
提取节点之后,我们要使用key()
函数获取要修改的键,这个函数会返回一个非常量的键。
需要注意的是,要将节点重新插会到map
时,我们需要在insert
中移动他们。因为extract
可避免不必要的拷贝和内存分配。还有一点就是,移动一个node_type
时,其不会让容器的任何值发生移动。
使用extract
方法提取的map
节点实际上非常通用。我们可以从一个map
实例中提取出来节点,然后插入到另一个map
中,甚至可以插入到multimap
实例中。这种方式在unordered_map
和unordered_multimap
实例中也适用。同样在set/multiset
和unordered_set/unordered_multiset
也适用。
为了在不同map
或set
结构中移动元素,键、值和分配器的类型都必须相同。需要注意的是,不能将map
中的节点移动到unordered_map
中,或是将set
中的元素移动到unordered_set
中。