��心删除容器中元素时的�P代器失效

Johnson — Wed, 14 Oct 2009 11:47:00 GMT

从一个简单的问题开始，删除数组中某个元素后�q�箋重复的元素，例如 1�Q?�Q?�Q?�Q?�Q?�Q?�Q?�Q?�Q? ---> 1�Q?2�Q?�Q?�Q?�Q?�?br>
考虑了几�U�，然后��开始动手写代码了：

#include <iostream>
#include <vector>
using namespace std;

int main(int argc, char* argv[])
{
   int a[] = {1, 1, 3, 3, 3, 2, 4, 1, 1, 1, 0};
   int size = sizeof(a)/sizeof(a[0]);

   vector<int> vec(a, a+size);

   vector<int>::iterator iter, end;
   int previous = vec[0];
   for (iter = vec.begin() + 1, end = vec.end(); iter != end; ++iter)
   {
      if(*iter == previous)
      {
         vec.erase(iter);
      }
      else
      {
         previous = *iter;
      }
   }

   for(iter = vec.begin(); iter != vec.end(); ++iter)
   {
      cout << *iter << endl;
   }

   return 0;
}

可是�~�译一下，出来一大堆error�Q�仔�l�看一下出错信息，哦，原来自己忘记了，erase容器中元素的时候，�q�代器会失效。。。顿时一�w�冷汗，自己�q�x��q�代容器的时候，一般都保存了容器的end元素�Q�要是此时�P代器失效。。�?/p>

Container<int>::iterator iter, end;
for (iter = container.begin() + 1, end = container.end(); iter != end; ++iter)

于是扑ֈ�收藏的Effective STL�Q�翻开条款9�Q�找��C��erase容器中元素的原则。以前曾�l�看�q�，不过�q�深日久�Q�早��忘得一�q�二净了。现在还是把要点�ȝ��一下，记在blog上，供以后参考�?br>
1. 对于兌��容器(如map, set, multimap,multiset)�Q�删除当前的iterator�Q�仅仅会使当前的iterator失效�Q�只要在erase�Ӟ��递增当前iterator卛_��。这是因为map之类的容器，使用了红黑树来实玎ͼ�插入、删除一个结点不会对其他�l�点造成影响�?br>

for (iter = cont.begin(); it != cont.end();)
{
   (*iter)->doSomething();
   if (shouldDelete(*iter))
      cont.erase(iter++);
   else
      ++iter;
}

因�ؓiter传给erase�Ҏ��的是一个副本，iter++会指向下一个元素�?br>
2. 对于序列式容�?如vector,deque)�Q�删除当前的iterator会��后面所有元素的iterator都失效。这是因为vetor,deque使用了连�l�分配的内存�Q�删除一个元素导致后面所有的元素会向前移动一个位�|�。还好erase�Ҏ��可以�q�回下一个有效的iterator�?br>

for (iter = cont.begin(); iter != cont.end();)
{
   (*it)->doSomething();
   if (shouldDelete(*iter))
      iter = cont.erase(iter);
   else
      ++iter;
}

3. 对于list来说�Q�它使用了不�q�箋分配的内存，�q�且它的erase�Ҏ��也会�q�回下一个有效的iterator�Q�因此上面两�U�方法都可以使用�?br>

最后给出开始那个问题的一个正��的实现�Q?/p>

#include <iostream>
#include <vector>
using namespace std;

int main(int argc, char* argv[])
{
   int a[] = {1, 1, 3, 3, 3, 2, 4, 1, 1, 1, 0};
   int size = sizeof(a)/sizeof(a[0]);

   vector<int> vec(a, a+size);

   vector<int>::iterator iter = vec.begin();
   int previous = *iter;
   ++iter;
   for (; iter != vec.end();)
   {
      if(*iter == previous)
      {
         iter = vec.erase(iter);
      }
      else
      {
         previous = *iter;
         ++iter;
      }
   }

   for(iter = vec.begin(); iter != vec.end(); ++iter)
   {
      cout << *iter << endl;
   }

   return 0;
}

PS. 不过实际上这个问题，用vector来实��C��是很适合�Q�因为每�ơ删除一个元素，都会引�vvector的一个resize操作。resize的时间复杂度是O(n)�Q�整个的resize操作要花费O(n^2)。最好是选择list最为容器，list最适合那些需要在容器中间做插入、删除的例子�?br>
PS2. 另外一个可能的�Ҏ��是remove_if + erase惯用法。remove_if的时间复杂度是O(n). erase的时间复杂度也是O(n)�Q�所以整个操作可以在O(n)旉��里完成�?当然�Q�对于这个例子，ShouldDeletePred比较隑ֆ�)

iterator new_end = remove_if(array.begin(), array.end(), ShouldDeletePred());
array.erase(new_end, array.end());

Johnson 2009-10-14 19:47 发表评论

嫩草影院久久99,久久精品国产黑森林,亚洲午夜福利精品久久

���心删除容器中元素时的�P代器失效

��心删除容器中元素时的�P代器失效