各種stlコンテナでのループ中の要素の削除 #C++

remove_if使えよと言われればそれまでですが普段使う感じだとループ中に消したりする事が多いので…

vector

vector<int> v;
v.push_back(10);
v.push_back(20);
v.push_back(30);

vector<int>::iterator it = v.begin();
while (it != v.end()) {
    if (*it == 20) {
        it = v.erase(it);
    } else ++it;
}

it = v.begin();
while (it != v.end()) {
    cout << *it << endl;
    ++it;
}

deque

deque<int> v;
v.push_back(10);
v.push_back(20);
v.push_back(30);

deque<int>::iterator it = v.begin();
while (it != v.end()) {
    if (*it == 20) {
        it = v.erase(it);
    } else ++it;
}

it = v.begin();
while (it != v.end()) {
    cout << *it << endl;
    ++it;
}

list

list<int> v;
v.push_back(10);
v.push_back(20);
v.push_back(30);

list<int>::iterator it = v.begin();
while (it != v.end()) {
    if (*it == 20) {
        it = v.erase(it);
    } else ++it;
}

it = v.begin();
while (it != v.end()) {
    cout << *it << endl;
    ++it;
}

map

map<int, int> v;
v[10] = 10;
v[20] = 20;
v[30] = 30;

map<int, int>::iterator it = v.begin();
while (it != v.end()) {
    if (it->second == 20) {
        v.erase(it++);
    } else ++it;
}

it = v.begin();
while (it != v.end()) {
    cout << it->first << " " << it->second << endl;
    ++it;
}

multimap

multimap<int, int> v;
v.insert(make_pair(10, 10));
v.insert(make_pair(20, 20));
v.insert(make_pair(30, 30));

multimap<int, int>::iterator it = v.begin();
while (it != v.end()) {
    if (it->second == 20) {
        v.erase(it++);
    } else ++it;
}

it = v.begin();
while (it != v.end()) {
    cout << it->first << " " << it->second << endl;
    ++it;
}

set

set<int> v;
v.insert(10);
v.insert(20);
v.insert(30);

set<int>::iterator it = v.begin();
while (it != v.end()) {
    if (*it == 20) {
        v.erase(it++);
    } else ++it;
}

it = v.begin();
while (it != v.end()) {
    cout << *it << endl;
    ++it;
}

とここまでやったけどmultisetとかは多分同じなので以下略…。。。

おおまかにvectorとかdequeとかシーケンスコンテナと呼ばれる奴はeraseの帰り値にイテレーターを返すのでそれが次のイテレーターになる。

mapとかsetとかは連想コンテナと言われる奴っぽく、eraseしたイテレーターは無効なアドレスを察すらしいので、 it++ を渡してイテレーターのコピーをしつつ要素を削除、かつイテレーターを勧める → コピーされたイテレーターを使って連想コンテナの要素を削除というノウハウがあるらしいここがわかりやすかった

たしかにこんな感じのコードを書くと違うっぽい。イテレーターの中身が書き変わってる。

cout << "OK" << endl;

{
    set<int> v;
    v.insert(10);
    v.insert(20);
    v.insert(30);

    set<int>::iterator it = v.begin();
    while (it != v.end()) {
        cout << *it << endl;

        if (*it == 20) {
            v.erase(it++);
        } else ++it;
    }
}

cout << endl;

cout << "NG" << endl;

{
    set<int> v;
    v.insert(10);
    v.insert(20);
    v.insert(30);

    set<int>::iterator it = v.begin();
    while (it != v.end()) {
        cout << *it << endl;

        if (*it == 20) {
            v.erase(it);
        }

        it++;
    }
}

まとめ

多分eraseでイテレーターが帰ってきたら it = v.erase(it); で、voidだったら if (XXX) { v.erase(++it); } else it++; で、みたいな感じの覚え方でざっくり大丈夫

全然気にしたことなかったけど各所でイテレーターとかループ回す時は前++のほうが効率がいいって書いてあるのでこれから気をつけます