[STL] 循环中erase

[STL] 循环中erase

sometimes,我们需要手写循环(相对于for_each)来erase容器内某些元素，新手经常会犯一些错误。这里总结一下比较常用的固定写法。

删除所有偶数项，并打印出删除的项

1. vector/queue
正确方法1：

void erase(vector<int> &v)
{
    for(vector<int>::iterator vi=v.begin();vi!=v.end();)
    {
        if(*vi % 2 == 0)
        {
            cout << "Erasing " << *vi << endl;
            vi = v.erase(vi);
        }
        else ++vi;
    }
}

正确方法2：

void erase2(vector<int> &v)
{
    for(vector<int>::reverse_iterator ri=v.rbegin();ri!=v.rend();)
    {
        if(*ri % 2 == 0)
        {
            cout << "Erasing " << *ri << endl;
            v.erase((++ri).base());
        }
        else ++ri;
    }
}

由于方法2是逆向删除，效率较高，推荐！

2.map/list
正确方法

void erase(map<int,int> &m)
{
    for(map<int,int>::iterator mi=m.begin();mi!=m.end();)
    {
        if(mi->second % 2 == 0)
        {
            cout << "Erasing " << mi->second << endl;
            m.erase(mi++);
        }
        else ++mi;
    }
}

2006-12-27 21:44 by eXile

// for vector, deque

template <class Container, class T>
inline
void vector_erase(Container & c, T const& t)
{
c.erase(std::remove(c.begin(), c.end(), t), c.end());
}

template <class Container, class Pred>
inline
void vector_erase_if(Container & c, Pred pred)
{
c.erase(std::remove_if(c.begin(), c.end(), pred), c.end());
}

// for list, set, map
template <class Container, class T>
void list_erase(Container & c, T const& t)
{
typename Container::iterator
b = c.begin(), e = c.end(), prev = b;
while (b != e)
{
++b;
if (*prev == t) c.erase(prev);
prev = b;
}
}

template <class Container, class Pred>
void list_erase_if(Container & c, Pred pred)
{
typename Container::iterator
b = c.begin(), e = c.end(), prev = b;
while (b != e)
{
++b;
if (pred(*prev)) c.erase(prev);
prev = b;
}
}

@eXile
如果需要打印出被删除的元素呢？

对于list,如果只要删除某个项，调用list.remove就可以了

2006-12-28 12:22 by eXile

1)打印出被删除的元素, 很简单
struct MyPred
{
bool operator()(int n) const
{
if(n%2 == 0) {cout << "Erasing " << n << endl; return true; }
else return false;
}
};

vector<int> v;
vector_erase_if(v, MyPred());
2)list提供了remove方法，但是set,map没有
（实际上这几行代码都是从STL的list源码中抄出来的，主要用于set 和map);