关闭

STL关联式容器中删除元素的方法和陷阱四

标签: iteratorpairdeletevectoriostreaminsert
1277人阅读 评论(0) 收藏 举报
分类:

 

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;

                   }

         }

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:55618次
    • 积分:745
    • 等级:
    • 排名:千里之外
    • 原创:5篇
    • 转载:19篇
    • 译文:2篇
    • 评论:7条
    最新评论