按索引删除一堆元素?

时间:2013-06-09 23:38:49

标签: c++ vector remove-if

我有一个

vector<int> myVector;

我有一个要删除的index列表:

vector<size_t> deleteIndex;

哪种策略删除这些索引最有效?

实际上,一个效率不高的解决方案是:

//> sort deleteindex
auto deleted= 0;
for(auto i=0;i<deleteIndex.size();i++ {
   myVector.erase(myVector.begin()+deleteIndex[i]-deleted);
   deleted++;
}

2 个答案:

答案 0 :(得分:1)

如果您被允许重新排序myVector,只需遍历要删除的项目,通过交换最后一个元素并将其弹出来删除它们。

如果您需要保留顺序,请对deleteIndex容器进行排序并执行单个有序传递以通过向前移动其他元素来移除元素。

答案 1 :(得分:1)

逐个擦除矢量中的元素效率非常低。这是因为对于每次擦除,它必须将所有元素向下复制一个,然后重新分配一个较小的向量。

相反,请使用erase-remove idiom。此过程将通过移动后续项目来替换先前的项目(它保持原始顺序)来删除项目。删除项目后,它将执行一次擦除(这只是列表的末尾)来重新分配一个小于n项的新向量(其中n是删除的项目数)。

示例实施:

template <class _FwdIt, class _FwdIt2>
_FwdIt remove_by_index(_FwdIt first, 
                       _FwdIt last,
                       _FwdIt2 sortedIndexFirst, 
                       _FwdIt2 sortedIndexLast)
{
  _FwdIt copyFrom = first;
  _FwdIt copyTo = first;
  _FwdIt2 currentIndex = sortedIndexFirst;

  size_t index = 0;
  for (; copyFrom != last; ++copyFrom, ++index)
  {
    if (currentIndex != sortedIndexLast &&
        index == *currentIndex)
    {
      // Should delete this item, so don't increment copyTo
      ++currentIndex;
      print("%d", *copyFrom);
    }
    else
    {
      // Copy the values if we're at different locations
      if (copyFrom != copyTo)
        *copyTo = *copyFrom;
      ++copyTo;
    }
  }
  return copyTo;
}

样本用法:

#include <vector>
#include <algorithm>
#include <functional>

int main(int argc, char* argv[])
{
  std::vector<int> myVector;

  for (int i = 0; i < 10; ++i)
    myVector.push_back(i * 10);

  std::vector<size_t> deleteIndex;
  deleteIndex.push_back(3);
  deleteIndex.push_back(6);

  myVector.erase(
    remove_by_index(myVector.begin(), myVector.end(), deleteIndex.begin(), deleteIndex.end()), 
    myVector.end());

  for (std::vector<int>::iterator it = myVector.begin();
       it != myVector.end(); ++it)
  {
    printf("%d ", *it);
  }

  return 0;
}

要点:https://gist.github.com/eleven41/5746079
在此测试:http://ideone.com/0qkDw5