Erase-remove惯用法

 

Erase-remove惯用法

分类： c/c++相关 2007-09-28 11:39  1140人阅读  评论(0)  收藏  举报

 

Erase-remove惯用法

Vector中包含8个元素,

现在要求删除是10的所有元素

10A , 10B都是代表10，只是为了区分，书写为10A，10B

10A0

10B0

10A1

10B1

10A2

21

10A3

22

 

很直观的写法

for  ( begin = vec.begin() ; begin != vec.end() ; ++begin )

{

if  ( 10 == *begin )

        {

                  vec.erase( begin ) ;

        }

}

但是运行之后，剩余的元素为

10B0

10B1

21

22

为什么会出现这种状况呢，我们分析一下

调用vec.erase删除10A0之后的情况如下

10B0

10A1

10B1

10A2

21

10A3

22

所有10A0之后的元素被逐次向前移动一个位置，即标记为红色的元素

此时begin指向10B0，那么begin++就为10A1，所以，10B0就被保留了下来（没有被删除）

同样的道理，删除10A1之后，经过begin++，那么10B1就又被保留了下来。

从而最后的结果变为

10B0

10B1

21

22

 

知道这个原因之后，我们来修改它

Erase的返回值是未被删除的第一个元素的iterator

for  ( begin = vec.begin() ; begin != vec.end() ; /* ++begin */ )

{

if  ( 10 == *begin )

        {

                begin = vec.erase( begin ) ;

        }

        else

        {

                 ++begin ;

        }

}

这样保证删除10A0之后，begin指向10B0，从而能够得到正确的结果。

不过这样显然不好，可以修改为如下形式

Vec.erase (         remove ( vec.begin() , vec.end() , 10 ) , vec.end() ) ;

 

更进一步，可以写为

CPred pred ;

Vec.erase (        remove_if ( vec.begin() , vec.end() ,

pred ) ,

vec.end() ) ;

 

这里pred是值传递的。

如果需求变为删除10的所有元素，并找出最大的元素。

那么可以这么写

Class CPred

{

Public:

              CPred( int nNum , int * pMax ) :

              M_nNum( mNum ) , m_pMax ( pMax ) {

                            *m_pMax = 0 ;

}

              Bool operator() ( int n )

              {

                            If ( n > *m_pMax )

                                          *m_pMax = n ;

                            If ( n == m_nNum )

                                          Return true ;

                            Else

                                          Return false ;

}

Private:

              Int m_nNum ;

              Int * m_pMax ;

} ;

调用者可以这么写

Int nMax = 0 ;

CPred pred( 10 , &nMax ) ;

Vec.erase (        remove_if ( vec.begin() , vec.end() ,

pred ) ,

vec.end() ) ;

调用之后, nMax就包含了最大的值。

 

因为是值传递，所以要想带回数据，必须使用指针。也可以声明为类CPred的静态变量。