使用erase删除std向量的元素对象:a)内存处理和b)更好的方法?

时间:2010-04-30 09:16:39

标签: c++ stdvector

我有vec_Aclass A的实例存储为:vec_A.push_back(A());

我想在稍后阶段删除向量中的一些元素并有两个问题:a)元素被删除为:vec_A.erase(iterator)是否需要添加任何其他代码以确保没有内存泄漏? 。

b)假设条件if(num <5)是num是特定numberList之一。鉴于此,有没有比我在下面说明的更好的方法来删除向量的元素?

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

class A {
      public:

             int getNumber();            
             A(int val);
             ~A(){};
      private:
              int num;
};

A::A(int val){
         num = val;
         };

int A::getNumber(){
    return num;
};

int main(){


    int i  =0;
    int num;
    std::vector<A> vec_A;
    std::vector<A>::iterator iter;

          for ( i = 0; i < 10; i++){
              vec_A.push_back(A(i));
          }
          iter = vec_A.begin();

          while(iter != vec_A.end()){
              std::cout <<  "\n --------------------------";
              std::cout <<  "\n Size before erase =" << vec_A.size();
              num = iter->getNumber() ;
              std::cout <<  "\n num = "<<num;
              if (num < 5){
                      vec_A.erase(iter);
                      }
              else{
                   iter++;
                   }

              std::cout <<  "\n size after erase =" << vec_A.size();
          }        


    std::cout << "\nPress RETURN to continue...";
    std::cin.get();

    return 0;
}

2 个答案:

答案 0 :(得分:4)

  

a)该元素被删除为:   vec_A.erase(iterator)有没有   我需要添加的其他代码   确定没有内存泄漏?

是的,这就是你需要做的。没有内存泄漏。由于您没有在堆上分配对象,因此当您执行vec_A.push_back(A())时,新对象将复制到向量中。擦除时,vector将负责删除元素。

  

假设条件if(num <5)是if   num是特定的numberList之一。   鉴于此,是否有更好的方法   删除矢量元素   我在下面说明了什么?

是的,你可以remove/erase成语。这是一个例子:

 class A
{
public:
    A(int n) : m_n(n)
    {
    }

    int get() const
    {
        return m_n;
    }
private:
    int m_n;
};

bool lessThan9(const A& a)
{
    return a.get() < 9;
}

//Or if you want for a generic number
struct Remover : public std::binary_function<A,int,bool>
{
public:
    bool operator()(const A& a, int n)const
    {
        return a.get() < n;
    }
};

int main()
{
    std::vector<A> a;
    a.push_back(A(10));
    a.push_back(A(8));
    a.push_back(A(11));
    a.push_back(A(3));

    a.erase(std::remove_if(a.begin(), a.end(), lessThan9), a.end());

    //Using the user-defined functor
    a.erase(std::remove_if(a.begin(), a.end(), std::bind2nd(Remover(), 9)), a.end());

    return 0;
}

答案 1 :(得分:1)

1)资源处理由类本身完成。类的析构函数负责确保没有内存泄漏。

2)从向量中删除元素最好从前到后进行:

for (std::vector<A>::reverse_iterator it = vec_A.rend(); it != vec_A.rbegin(); --it)
{
  if (it->getNumber() < 5) {vec_A.erase(it.base());}
}