Erase-remove惯用法
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的静态变量。