web-dev-qa-db-ja.com

ベクターから要素を消去する

Eraseメソッドを使用してベクターから要素をクリアしたい。しかし、ここでの問題は、要素がベクトル内で一度だけ発生することが保証されていないことです。複数回存在する可能性があり、それらすべてをクリアする必要があります。私のコードは次のようなものです：

void erase(std::vector<int>& myNumbers_in, int number_in)
{
    std::vector<int>::iterator iter = myNumbers_in.begin();
    std::vector<int>::iterator endIter = myNumbers_in.end();
    for(; iter != endIter; ++iter)
    {
        if(*iter == number_in)
        {
            myNumbers_in.erase(iter);
        }
    }
}

int main(int argc, char* argv[])
{
    std::vector<int> myNmbers;
    for(int i = 0; i < 2; ++i)
    {
        myNmbers.Push_back(i);
        myNmbers.Push_back(i);
    }

    erase(myNmbers, 1);

    return 0;
}

繰り返し処理中にベクターの末尾を変更しているため、このコードは明らかにクラッシュします。これを達成する最良の方法は何ですか？つまりベクターを複数回繰り返したり、ベクターのコピーをもう1つ作成せずにこれを行う方法はありますか？

c++vectorstlerase
91
Naveen

remove/erase idiom を使用します。

std::vector<int>& vec = myNumbers; // use shorter name
vec.erase(std::remove(vec.begin(), vec.end(), number_in), vec.end());

起こるのは、removeが、削除される値とは異なる要素を圧縮することです（number_in）vectorの先頭で、その範囲の後の最初の要素に反復子を返します。次に、eraseはこれらの要素を削除します（値は指定されていません）。

148
Motti

Eraseを呼び出すと、イテレータが無効になります：

void erase(std::vector<int>& myNumbers_in, int number_in)
{
    std::vector<int>::iterator iter = myNumbers_in.begin();
    while (iter != myNumbers_in.end())
    {
        if (*iter == number_in)
        {
            iter = myNumbers_in.erase(iter);
        }
        else
        {
           ++iter;
        }
    }

}

または、 std :: remove_if をファンクターとstd :: vector :: eraseとともに使用できます。

struct Eraser
{
    Eraser(int number_in) : number_in(number_in) {}
    int number_in;
    bool operator()(int i) const
    {
        return i == number_in;
    }
};

std::vector<int> myNumbers;
myNumbers.erase(std::remove_if(myNumbers.begin(), myNumbers.end(), Eraser(number_in)), myNumbers.end());

この場合、独自のファンクターを作成する代わりに、 std :: remove を使用できます。

std::vector<int> myNumbers;
myNumbers.erase(std::remove(myNumbers.begin(), myNumbers.end(), number_in), myNumbers.end());
47
dalle

  1. インデックスアクセスを使用して反復できます。

  2. O（n ^ 2）の複雑さを避けるために、2つのインデックスを使用できます。i-現在のテストインデックス、j-次のアイテムを格納するインデックスと、サイクルの最後にベクトルの新しいサイズ。

コード：

void erase(std::vector<int>& v, int num)
{
  size_t j = 0;
  for (size_t i = 0; i < v.size(); ++i) {
    if (v[i] != num) v[j++] = v[i];
  }
  // trim vector to new size
  v.resize(j);
}

そのような場合、イテレーターの無効化はなく、複雑さはO（n）であり、コードは非常に簡潔であり、ヘルパークラスを作成する必要はありませんが、ヘルパークラスを使用すると、より柔軟なコードが得られます。

このコードはeraseメソッドを使用しませんが、タスクを解決します。

純粋なstlを使用すると、次の方法でこれを行うことができます（これはMottiの答えに似ています）。

#include <algorithm>

void erase(std::vector<int>& v, int num) {
    vector<int>::iterator it = remove(v.begin(), v.end(), num);
    v.erase(it, v.end());
}
12
sergtk

これを行う理由に応じて、 std :: set を使用する方がstd :: vectorよりも良いアイデアかもしれません。

各要素が1回だけ出現することを許可します。複数回追加する場合は、とにかく消去するインスタンスは1つだけです。これにより、消去操作が簡単になります。消去操作の時間の複雑さはベクターの場合よりも低くなりますが、要素の追加はセット上で遅くなるため、あまり利点がない場合があります。

エレメントがベクターに追加された回数やエレメントが追加された順序に関心がある場合、これはもちろん機能しません。

3
Laserallan