从std :: list中删除具有特定值的元素

时间:2009-03-24 15:27:44

标签: c++ list stl

我需要从std :: list中删除具有特定值的元素。使用list<int>我使用了remove()方法。

现在我有list<CMyClass>所以我认为我应该使用remove_if()但它的谓词只需要一个参数 - 要测试的元素。

如何编写一个函数foo(const CMyClass &Bad),它从列表中删除所有等于Bad的元素?

由于

PS

struct CMyClass {
    void *Ptr;
    int Var;
}

bool is_equal(const CMyClass &A, const CMyClass &B)
{
    if (A.Ptr == B.Prt and A.Var == B.Var)
        return true;
    else
        return false;
}

7 个答案:

答案 0 :(得分:15)

您的类必须在ClassName中实现operator ==

bool operator == ( const Class& rhs );

然后你可以使用

list.remove( Bad )

如果你的类有合理的运算符==(不仅仅是删除) - 比list :: remove更适合你的解决方案。如果operator ==仅用于list :: remove,则最好使用remove_if。

在以下示例中,演示了list :: remove和list :: remove_if。

struct Class
{
    int a_;
    int b_;

    Class( int a, int b ):
        a_( a ),
        b_( b )
    {}

    bool operator == (const Class &rhs)
    {
        return (rhs.a_ == a_ && rhs.b_ == b_);
    }

    void print()
    {
        std::cout << a_ << " " << b_ << std::endl;
    }
};

bool isEqual( Class lhs, Class rhs )
{
    return (rhs.a_ == lhs.a_ && rhs.b_ == lhs.b_);
}

struct IsEqual
{
    IsEqual( const Class& value ):
        value_( value )
    {}

    bool operator() (const Class &rhs)
    {
        return (rhs.a_ == value_.a_ && rhs.b_ == value_.b_);
    }

    Class value_;
};

int main()
{
    std::list<Class> l;

    l.push_back( Class( 1, 3 ) );
    l.push_back( Class( 2, 5 ) );
    l.push_back( Class( 3, 5 ) );
    l.push_back( Class( 3, 8 ) );

    Class bad( 2, 5 );

    std::cout << "operator == " << std::endl;
    l.remove( bad );
    std::for_each( l.begin(), l.end(), std::mem_fun_ref( &Class::print ) );

    std::cout << "binary function predicat" << std::endl;
    l.push_back( Class( 2, 5 ) );
    l.remove_if( std::bind2nd( std::ptr_fun(isEqual), bad ) );
    std::for_each( l.begin(), l.end(), std::mem_fun_ref( &Class::print ) );


    std::cout << "functor predicat" << std::endl;
    l.push_back( Class( 2, 5 ) );
    l.remove_if( IsEqual( bad ) );
    std::for_each( l.begin(), l.end(), std::mem_fun_ref( &Class::print ) );

    return 0;
}

答案 1 :(得分:6)

根据谓词删除所有元素:

struct Pred
{
   bool operator()(int i) const
   {
      return i == 42;
   }
};

std::list<int> l = ...;
l.remove_if(Pred());

或删除值为42的所有元素:

std::list<int> l = ...;
l.remove(42);

或CMyClass列表:

struct Pred
{
   bool operator()(const CMyClass& item) const
   {
      return item.GetSomething() == 42 && item.GetSomethingElse() == 314159;
   }
};

std::list<CMyClass> l = ...;
l.remove_if(Pred());

答案 2 :(得分:3)

创建一个接受错误值的类并将其存储在成员变量中。然后实现

bool operator()(CMyClass const &currVal) { 
  return (currVal is equal to this->bad);
}

将此对象的实例传递给remove_if

答案 3 :(得分:2)

如果条件匹配,您可以简单地迭代元素,比较和擦除,例如:

for (std::list<string>::iterator i = mylist.begin(), e = mylist.end(); i != e; )
{
    if (*i == "foobar")
       i = mylist.erase(i);
    else
       ++i;
}

答案 4 :(得分:1)

实现operator ==(const CMyClass&amp;)。

e.g:

#include <list>
#include <iostream>
#include <iterator>
using namespace std;

class CMyClass {
public:
  CMyClass(const int d) : data(d) {}
  bool operator==(const CMyClass &rhs) { return data == rhs.data; }
friend
  ostream& operator<<(ostream &ost, const CMyClass &rhs) { return ost << rhs.data; }
private:
  int data;
};


int main(int, char **) {
  list<CMyClass> li;
  CMyClass a(1);
  CMyClass b(8);
  li.push_back(a);
  li.push_back(b);
  copy(li.begin(), li.end(), ostream_iterator<CMyClass>(cout,"\n"));
  li.remove(a);
  copy(li.begin(), li.end(), ostream_iterator<CMyClass>(cout,"\n"));
  return 0;
}

结果:

1 8 8

答案 5 :(得分:0)

using namespace std;

class Foo
{
    public:
        Foo(int newVal){value = newVal;}
        int value;
};

class checkEqual : public binary_function<Foo, Foo, bool>
{
    public:
        bool operator()(const Foo &inputOne, const Foo &inputTwo) const
        {   return inputOne.value == inputTwo.value;}
            // implement your comparison here
};

int main(int count, char** args)
{
    list<Foo> myList;

    for(int x = 0; x < 10; ++x)
        myList.push_back(Foo(x));

    Foo bad(5);

    myList.remove_if(bind2nd(checkEqual(), bad));    

    return 0;
}

或者如果你想重载==你可以这样做:

class Foo
{
    public:
        Foo(int newVal){value = newVal;}
        int value;
        bool operator==(const Foo &other) const
            {    return this->value == other.value;}
                 // your comparison here
};

然后:

myList.remove(bad);

答案 6 :(得分:-1)

尝试使用remove_if方法

http://www.sgi.com/tech/stl/remove_if.html