在迭代相同的向量时删除向量中的元素

时间:2012-03-29 14:13:25

标签: c++ vector

  

可能重复:
  Erasing from a std::vector while doing a for each?

我正在尝试根据此算法实现顶点着色;

/*
Given G=(V,E):
Compute Degree(v) for all v in V.
Set uncolored = V sorted in decreasing order of Degree(v).
set currentColor = 0.
while there are uncolored nodes:
   set A=first element of uncolored
   remove A from uncolored
   set Color(A) = currentColor
   set coloredWithCurrent = {A}
   for each v in uncolored:
      if v is not adjacent to anything in coloredWithCurrent:
         set Color(v)=currentColor.
         add v to currentColor.
         remove v from uncolored.
      end if
   end for
   currentColor = currentColor + 1.
end while
*/

我不明白“将v添加到currentColor”。但是我认为,这意味着将currentColor与v协调。因此什么是“set”?无论如何,问题是在迭代它时在向量中擦除元素。这是代码。

    vector<struct Uncolored> uc;
    vector<struct Colored> c;   

    int currentColor = 0;
    struct Colored A;
    struct Colored B;

    vector<struct Uncolored>::iterator it;
    vector<struct Uncolored>::iterator it2;
    vector<struct Colored>::iterator it3;

    for(it=uc.begin();it<uc.end();it++){

        A.id = (*it).id;        
        uc.erase(uc.begin());
        A.color = currentColor;
        c.push_back(A);

        for(it2=uc.begin();it2<uc.end();it2++) {
            it3=c.begin();
            while(it3 != c.end()) {
                if( adjacencyMatris[(*it2).id][(*it3).id] == 0 ) {
                    B.id = (*it2).id;       
                    it2 = uc.erase(it2);
                    B.color = currentColor;
                    c.push_back(B);
                }
                it3++;
            }
        }
        currentColor = currentColor + 1;
    }

我认为it2 = uc.erase(it2);行已经普遍使用,但它会产生运行时错误。

4 个答案:

答案 0 :(得分:37)

在行中:

it2 = uc.erase(it2);

迭代器it2指向的元素将从向量中移除,元素在内存中移位以填充使it2无效的间隙。 it2获取一个新值,现在指向移除的一个或向量末尾之后的第一个元素(如果删除的元素是最后一个元素)。这意味着在擦除元素后,您不应该前进it2。提议的remove-erase idiom的替代方法是一个简单的技巧:

for(it2 = uc.begin(); it2 != uc.end();)
{
   ...   
   if(...)
   {
      it2 = uc.erase(it2); 
   }
   else
   {
      ++it2;
   }
   ...
}

您可以阅读有关此here的更多信息。

修改 关于你的评论,你可以使用一个标志来传递一个元素是否被删除的信息,你可以从内循环中检查它:

for(it2=uc.begin(); it2 != uc.end();)
{
   bool bErased = false;

   for(it3 = c.begin(); it3 != c.end(); ++it3)
   {
      if(adjacencyMatris[(*it2).id][(*it3).id] == 0 )
      {
         B.id = (*it2).id;
         it2 = uc.erase(it2);
         bErased = true;
         B.color = currentColor;
         c.push_back(B);
         break;
      }
   }

   if(!bErased)
      ++it2;
}

uc删除元素后,您需要从内循环中删除。在外部循环的下一次迭代中,您将能够通过有效的迭代器访问uc中的下一个元素。

答案 1 :(得分:2)

vector::erase()invalidate iterators指向向量。

  

这会使所有迭代器和对位置(或第一个)及其后续元素的引用无效。

你需要将erase的结果添加到迭代器中(它将指向一个擦除后的元素),然后使用它。请注意

for(it=uc.begin();it<uc.end();++it){ 
  A.id = (*it).id;         
  uc.erase(uc.begin()); 
  ...
}

迭代器ituc.erase之后无效,因此后续++和use可能会导致运行时错误。

同样,即使您将擦除结果分配给it2,通话也会使it无效,但不会更改。

您最好的选择是在每个erase()之后从头开始重新启动算法,或者如果您可以更改它以便它可以从erase返回的迭代器继续,那么执行此操作获得一些效率。

答案 2 :(得分:2)

不要使用iterator类型,而是将索引存储到vector中。当你需要一个迭代器 - 也许是为了传递给erase - 你可以说begin() + myIndex来生成一个迭代器。

这也使循环看起来更熟悉,例如。

for(ind=0; ind < uc.size(); ind++) {

答案 3 :(得分:0)

您遇到了运行时错误,因为it2 = uc.erase(it2);返回最后一个删除元素后的迭代器,因此it2++中的for(it2=uc.begin();it2<uc.end();it2++)超出了最后一个元素。

尝试更改if in:

if( adjacencyMatris[(*it2).id][(*it3).id] == 0 ) {
    B.id = (*it2).id;       
    uc.erase(it2);
    B.color = currentColor;
    c.push_back(B);
    break;
}