这是在std :: vector中存储,迭代和删除指针的好方法吗?

时间:2012-09-10 11:01:25

标签: c++ pointers functor stdvector erase-remove-idiom

#include <iostream>
#include <cstdlib>
#include <ctime>
#include <algorithm>
#include <vector>
using namespace std;

struct delete_ptr
{
    template<typename T>
    void operator()(T*& t)
    {
        delete t;
        t = 0;
    }
};

struct is_null_ptr
{
    template<typename T>
    bool operator()(T*& t)
    {
        return t == 0;
    }
};

struct A
{
    static void removeDead(A*& a)
    {
        if(a and a->dead)
            delete_ptr()(a);
    }

    static void killSome(A* a)
    {
        if(a and a->isDead() == false and rand()%100 == 0)
        {
            static int counter = 0;
            cout << "Kill___" << ++counter << endl;
            a->kill();
        }
    }

    static void reviveSome(A* a)
    {

        if(a and a->isDead() and rand()%3 == 0)
        {
            static int counter = 0;
            cout << "Revive___" << ++counter << endl;
            a->revive();
        }
    }

    A():dead(false)
    {

    }

    virtual ~A()
    {
        static int counter = 0;
        cout << "Dtor___" << ++counter << endl;
    }

    bool isDead(){return dead;}
    void kill(){dead = true;}
    void revive(){dead = false;}

    bool dead;
};

int main()
{
    srand(time(0));
    vector<A*> as;
    for(int i = 0; i < 200; ++i)
    {
        A* a = new A;
        as.push_back(a);
    }


    for_each(as.begin(),as.end(),A::killSome);
    for_each(as.begin(),as.end(),A::reviveSome);

    for_each(as.begin(),as.end(),A::removeDead);
    as.erase( std::remove_if(as.begin(),as.end(),is_null_ptr()),as.end());
    cout << as.size() << endl;

    for_each(as.begin(),as.end(),delete_ptr());
    as.clear();

    return 0;
}

它分配它们,并输出正确的输出,但我不确定这是我正在做的正确的事情。我只是试图在向量中使用指针并在发生某种情况时删除它们,而不使用boost或c ++ 11。 那你觉得怎么样?

1 个答案:

答案 0 :(得分:1)

由于当前STL(auto_ptr)中唯一的智能指针不能用于容器中,我会说在给定条件下你的方式很好。

但您可以考虑实现自己的unique_ptr或shared_ptr。

PS:使用指针而不是容器中的实际对象有很多原因,一个是多态。另一个是实际对象已经存储在其他地方(想想已存储对象的索引结构)。