今天同学群里面讨论了这样一段代码，说是产品出了 bug，现场急着修复。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
| #include <iostream>
#include <map>
#include <string>
int main()
{
std::cout << "hello world" << std::endl;
std::map<int, std::string> mapPeople;
mapPeople[1] = "hexu1";
mapPeople[2] = "hexu2";
std::map<int, std::string>::iterator iter = mapPeople.begin();
for (; iter != mapPeople.end(); iter++)
{
if (iter->first == 2)
{
std::cout << "id : " << iter->first << ", name : " << iter->second << std::endl;
mapPeople.erase(iter++);
std::cout << (iter != mapPeople.end()) << std::endl;
}
}
return 0;
}
|
说是第一个 for 循环会崩溃，第二个 for 循环可以通过。经过分析，主要问题其实是出在对迭代器和 erase 的不熟悉上，导致错误的使用，这个程序的错误很明显，就是在 erase 这条分支中，iter++了两次。那么如何写一个正确的 remove 函数呢？代码如下：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
| #include <iostream>
#include <map>
#include <string>
int main()
{
std::cout << "hello world" << std::endl;
std::map<int, std::string> mapPeople;
mapPeople[1] = "hexu1";
mapPeople[2] = "hexu2";
mapPeople[3] = "hexu3";
mapPeople[4] = "hexu4";
std::map<int, std::string>::iterator iter = mapPeople.begin();
while (iter != mapPeople.end())
{
if (iter->first == 2)
{
std::cout << "id : " << iter->first << ", name : " << iter->second << std::endl;
iter = mapPeople.erase(iter);
continue;
}
iter++;
}
return 0;
}
|
另外，碰到删除的时候，要小心，可以这么写：iter = mapPeople.erase(iter);，也可以这么写：mapPeople.erase(iter++);（必须在删除前给迭代器++，否则迭代器会失效，这个问题那位程序员倒是考虑到了）。
最后我想说的一点是，根据指定 key 删除 map 中的元素，这个功能在其他语言中都有库函数可以调用，而 C++却没有，还要自己动手写，而且需要对 map 的原理有一定的了解。这就从侧面反映了 C++可能并不是一个高效的生产工具。