在
STL
(标准模板库)中经常会碰到要删除容器中部分元素的情况,本人在编程中就经常编写这方面的代码,在编码和测试过程中发现在
STL
中删除容器有很多陷阱,网上也有不少网友提到如何在
STL
中安全删除元素这些问题。
上一篇文章主要讨论序列式容器
vector
、
list
中安全删除元素的方法和可能会遇到的陷阱,这一次讨论在
map
(
multimap
)容器中如何安全的删除和插入元素。
map
(
multimap
)容器为关联式容器,是编程中经常使用的容器,有键值(
key
)和实值(
value
),又称字典、映射表。
你能看出以下代码有什么问题?
例
1
:
#pragma warning (disable : 4786)
#include <iostream>
#include <map>
using namespace std;
void main() {
map< int, int* > mapInt;
for ( int i = 0; i < 5; i++ ) {
mapInt[ i ] = new int( i );
}
//
再插入键值为
2
的元素
mapInt[ 2
] = new int( 2 );
//
做一些操作
//
删除内存。
map< int, int* >::iterator itMap = mapInt.begin();
for ( ; itMap != mapInt.end();
++itMap ) {
delete itMap->second;
}
}
例
2
:
void main() {
map< int, int* > mapInt;
for ( int i = 0; i < 5; i++ ) {
mapInt.insert( make_pair( i, new int( i ) ) );
}
//
再插入键值为
2
的元素
mapInt.insert( make_pair( 2, new int( 2 ) ) );
//
做一些操作
//
删除内存。
map< int, int* >::iterator itMap = mapInt.begin();
for ( ; itMap != mapInt.end();
++itMap ) {
delete itMap->second;
}
}
例
3
:
void main() {
map< int, int > mapInt;
for ( int i = 0; i < 5; i++ ) {
mapInt.insert( make_pair( i,
i ) );
}
mapInt.insert( make_pair( 5,
2 ) );
//
删除所有实值为
2
的元素
map< int, int >::iterator itMap = mapInt.begin();
for ( ; itMap != mapInt.end();
++itMap ) {
if (
itMap->second == 2 ) {
mapInt.erase( itMap );
}
}
}
分析:
例
1
将导致内存泄露,因为
mapInt[ 2
] = new int( 2 );
这条语句把原来键值为
2
的元素的实值指针覆盖了,原来的指针就成为野指针,导致内存泄露。
例
2
也将导致内存泄露,因为
mapInt.insert( make_pair( 2, new int( 2 ) ) );
这条语句因为键值为
2
的元素已经存在,导致插入元素失败,所以指向刚分配的内存的指针成为野指针,导致内存泄露。
map
容器插入元素的方法。可以调用
map
容器的
insert
成员函数插入元素,或者直接用
map
容器的下标运算式赋值,但这里有一个地方要注意,如果实值为指针的话,插入重复键值的元素时,会导致内存泄露。所以对于插入元素时,必须检查是否插入成功。
正确的方法:
void main() {
map< int, int* > mapInt;
bool bRet;
for ( int i = 0; i < 5; i++ ) {
int* pI = new int( i );
bRet = mapInt.insert( make_pair( i, pI ) ).second;
if ( !bRet ) {
//
插入元素不成功。
delete pI;
}
}
//
再插入键值为
2
的元素
int* pI = new int( 2 );
bRet = mapInt.insert( make_pair( 2, pI ) ).second;
if ( !bRet ) {
//
插入元素不成功。
delete pI;
}
//
做一些操作
//
删除内存。
map< int, int* >::iterator itMap = mapInt.begin();
for ( ; itMap != mapInt.end();
++itMap ) {
delete itMap->second;
}
}
例
3
将导致程序未定义的错误,在
windows
中即是访问非法内存,程序当掉。因为
mapInt.erase( itMap );
调用后
itMap
迭代器已无效了,所以当执行
++itMap
时,访问非法内存,导致程序当掉。
如果
erase
()总是返回下一元素的位置,那就可以像在
vector
容器中删除元素一样,如:
//
删除所有实值为
2
的元素
map< int, int >::iterator itMap = mapInt.begin();
for ( ; itMap != mapInt.end();
) {
if (
itMap->second == 2 ) {
itMap = mapInt.erase( itMap );
}
else {
++itMap;
}
}
但是,注意,以上的方式只在
vc
使用
P.J.STL
中才能编译通过,而使用
SGI STL
库则编译不过,因为
SGISTL
库在设计中考虑到如果用户不需要这一特性,就会损失性能,因此否决了这种想法。所以要保证可移植性,最好采用下面的方法:
//
删除所有实值为
2
的元素
map< int, int >::iterator itMap = mapInt.begin();
for ( ; itMap != mapInt.end();
) {
if (
itMap->second == 2 ) {
//
itMap++
将使
itMap
指向下一个元素,但返回原迭代器的副本,所以
//
erase
()被调用时,
itMap
不指向将要被删除的元素
mapInt.erase( itMap++ );
}
else {
++itMap;
}
}