STL中remove和remove_if用法

 

remove 和 remove_if方法都是包含在<algorithm>頭文件當中的，先來看看原型：

// TEMPLATE FUNCTION remove
template<class _FI, class _Ty> inline
    _FI remove(_FI _F, _FI _L, const _Ty& _V)
    {_F = find(_F, _L, _V);
    if (_F == _L)
        return (_F);
    else
        {_FI _Fb = _F;
        return (remove_copy(++_F, _L, _Fb, _V)); }}
        // TEMPLATE FUNCTION remove_if
template<class _FI, class _Pr> inline
    _FI remove_if(_FI _F, _FI _L, _Pr _P)
    {_F = find_if(_F, _L, _P);
    if (_F == _L)
        return (_F);
    else
        {_FI _Fb = _F;
        return (remove_copy_if(++_F, _L, _Fb, _P)); }}

這兩個方法的前兩個參數均相同，第三個參數不同。
remove的第三個參數是常量

如：

去除vector中為3的元素：

int main(int argc, char* argv[])
{
    std::vector<int> vec;
    vec.push_back(2);
    vec.push_back(3);
    vec.push_back(4);
    vec.push_back(6);
    vec.push_back(8);
    vec.push_back(12);
    vec.push_back(22);
    vec.push_back(33);
                
                //從前往后找出3的元素，並從vector中erase
                //remove第三個參數為常數的情況
    vec.erase( std::remove(vec.begin(),vec.end(),3), vec.end() );

    std::vector<int>::iterator iter = vec.begin();
    for (iter = vec.begin(); iter != vec.end(); iter++)
    {
        printf("%d\n",(*iter));
    }

    return 0;
}

該方法std::remove(vec.begin(),vec.end(),3); 不會改變容器大小，並返回容器末端的最后一個元素.

所以單看這條語句，如打印整個容器，得到的將是：

2 4 6 8 12 22 33 33

刪除的元素后面的全部往前移，並最后一個位置返回之前的最后一個元素。故最后的元素多打印了一次。

該方法結合erase方法：vec.erase( std::remove(vec.begin(),vec.end(),3), vec.end() ); 就是將返回的多出來的最后一個元素從容器中erase掉。

從而得到最后正確的結果：

2 4 6 8 12 22 33

*********************************************

remove_if方法的第三個參數是一個過濾條件，即我們要使用何種清除規則來清楚容器中的元素

如，我們要刪除容器中所有為3的倍數的元素:

 

//該方法實現對3的倍數的元素的過濾
//參數是容器中的元素，注意返回值類型
bool RemoveItem(int item)
{
    if (item%3 == 0)
    {
        return true;
    }
    else
    {
        return false;
    }
}

int main(int argc, char* argv[])
{
    std::vector<int> vec;
    vec.push_back(2);
    vec.push_back(3);
    vec.push_back(4);
    vec.push_back(6);
    vec.push_back(8);
    vec.push_back(12);
    vec.push_back(33);
    vec.push_back(22);
//第三個參數是使用函數指針的方法調用，參數由系統自動傳遞
//返回值為真則清除，假則不清除
   vec.erase( std::remove_if(vec.begin(),vec.end(),RemoveItem), vec.end() );

    std::vector<int>::iterator iter;
    for (iter = vec.begin(); iter != vec.end(); iter++)
    {
        printf("%d\n",(*iter));
    }

    return 0;
}

打印結果為：

2 4 8 22