从集合中删除元素的问题

时间:2011-08-06 13:32:29

标签: c++ set erase

我在从集合中删除元素时遇到了麻烦。我从BUILD FAILED获得了:{/ p>

n2Ar.erase(it);
n3Ar.erase(it);

其中it是从find()函数收到的指针:例如it = n2Ar.find(*i);

该计划的整个清单:

#include <stdio.h>
#include <iostream>
#include <vector>
#include <set>
#include <algorithm>

using namespace std;

#define TESTING_FILE_IN
//#define TESTING_FILE_OUT
//#define DEBUG
//#define SHOW_TIMING

int outputSet(int i) {
    cout << i << endl;
}

/*
 * 
 */
int main() {

    int n1, n2, n3;
    set<int> list, n1Ar, n2Ar, n3Ar;
    set<int>::iterator it;

    scanf("%d", &n1);
    scanf("%d", &n2);
    scanf("%d", &n3);

    int val = 0;

    // Getting lists of voters
    for (unsigned i = 0; i < n1; i++) {
        cin >> val;
        n1Ar.insert(val);
    }

    for (unsigned i = 0; i < n2; i++) {
        cin >> val;
        n2Ar.insert(val);
    }

    for (unsigned i = 0; i < n3; i++) {
        cin >> val;
        n3Ar.insert(val);
    }

    // Processing lists

    for (set<int>::iterator i = n1Ar.begin(); i != n1Ar.end(); ++i) {
        it = n2Ar.find(*i);

        if (it != n2Ar.end()) {
            list.insert(*i);
            n1Ar.erase(i);
            n2Ar.erase(it);

        } else {

            it = n3Ar.find(*i);
            if (it != n3Ar.end()) {
                list.insert(*i);
                n1Ar.erase(i);
                n3Ar.erase(it);
            }
        }
    }

    // Outputting the final list
    cout << list.size() << endl;
    for_each(list.begin(), list.end(), outputSet);

    return 0;
}

我希望你能帮助我理解我在这里做错了什么。我只是从C ++开始。

3 个答案:

答案 0 :(得分:2)

您的代码中存在两个问题。

首先,您需要在以下函数中返回一个值,或者只是让它返回void。

// you should return a value here or make it return void
int outputSet(int i)
{
    cout << i << endl;
}

第二个,一旦删除当前迭代,for循环的以下迭代中的迭代器 无效 。一旦删除了一个元素,它的迭代器i也会失效,以便基于++ i的以下迭代器;

你会得到运行时错误,因为迭代器我现在指向你需要以某种方式“重置”它。

MSVC实施

for (set<int>::iterator i = n1Ar.begin(); i != n1Ar.end(); ++i) {
        it = n2Ar.find(*i);

        if (it != n2Ar.end()) {
            list.insert(*i);
            // the following iterators become invalidated after the
            // current one is removed. You need reset it like
            // i = n1Ar.erase(i);
            n1Ar.erase(i);
            n2Ar.erase(it);

        } else {

            it = n3Ar.find(*i);
            if (it != n3Ar.end()) {
                list.insert(*i);
                // the following iterators become invalidated after the
                // current one is removed. You need reset it like
                // i = n1Ar.erase(i);
                n1Ar.erase(i);
                n3Ar.erase(it);
            }
        }
    }

编辑:请注意,从set :: erase()返回一个新的迭代器不是标准方式。这主要是为了表现。

更便携的解决方案

基本思想是在删除当前迭代器之前正确设置下一个迭代器。

   set<int>::iterator i = n1Ar.begin();

   while (i != n1Ar.end())
   {
      it = n2Ar.find(*i);
      if (it != n2Ar.end())
      {
         // the trick is to use "i++" where i is incremented by one while "old" i
         // is removed.
         list.insert(*i);
         n1Ar.erase(i++);
         n2Ar.erase(it);
      }
      else
      {    
         it = n3Ar.find(*i);
         if (it != n3Ar.end())
         {
            list.insert(*i);
            n1Ar.erase(i++);
            n3Ar.erase(it);
         }
         else
         {
            ++i;
         }
      }
   }

答案 1 :(得分:0)

n1Ar.erase(i);

std :: set :: erase函数使迭代器i无效并导致问题。请考虑更改以下内容:

for (set<int>::iterator i = n1Ar.begin(); i != n1Ar.end(); ++i) {
        it = n2Ar.find(*i);

        if (it != n2Ar.end()) {
            list.insert(*i);
            i = n1Ar.erase(i);
            if(i == n1Ar.cend())
                break;
            n2Ar.erase(it);

        } else {

if(i == n1Ar.cend()) break;检查有助于确保无效的迭代器不会破坏循环。

答案 2 :(得分:0)

  1. 擦除方法使迭代器i无效。
  2. Erase方法不返回迭代器。
  3. 编辑:重复Eric的想法,您可以使用以下代码:

    for (set<int>::iterator i = n1Ar.begin(); i != n1Ar.end(); )
        if ( n2Ar.erase(*i) || n3Ar.erase(*i) ) {
            list.insert(*i);
            n1Ar.erase(i++);
        } else i++;
    

    此问题也可以使用标准算法解决。但这种解决方案似乎效率较低:

    set<int> tmp;
    std::set_union( n2Ar.begin(), n2Ar.end(),
        n3Ar.begin(), n3Ar.end(), std::inserter(tmp,tmp.begin()) );
    std::set_intersection( n1Ar.begin(), n1Ar.end(),
        tmp.begin(), tmp.end(), std::inserter(list,list.begin()) );
    

    最后我建议使用stl作为输出(你必须包含迭代器库):

    std::copy( list.begin(), list.end(), std::ostream_iterator<int>(std::cout,"\n"));