[C++] vector删除指定元素

背景

最近在工作中遇到一个问题，问题可简化如下：

在推荐系统中，对某个user的召回候选item集合进行打分排序，item集合存放在一个召回列表vector中。需要实现的功能为，在召回item列表送入模型打分之前，先做过滤。即对于user已经拥有的item，不再将这些item送入模型进行打分。

例如：user1已经拥有item1，item2。则需要将召回列表vector中的item1，item2过滤掉。

解决方案

int32_t filter_number = 0;
std::set<int64_t> user_already_ow_item_set;
vector<ItemInfo>::iterator it;
for (it = item_recall_list.begin(); it != item_recall_list.end();) {
    if (user_already_own_item_set.find((*it).item_id) != user_already_own_item_set.end()) {
      it = item_recall_list.erase(it); // 不可写为item_recall_list.erase(it);
      filter_number++;
    } else {
      ++it;	// 
    }
}

知识点

C++ vector中，实际删除元素使用的是容器vecrot中std::vector::erase()方法；std::remove()并不删除元素，因为容器的size()没有变化，只是元素的替换。

• std::vector::erase() 函数原型：
  iterator erase (iterator position);　　//删除指定元素
  iterator erase (iterator first, iterator last);　　//删除指定范围内的元素

  返回值：指向删除元素(或范围)的下一个元素。（An iterator pointing to the new location of the element that followed the last element erased by the function call. This is the container end if the operation erased the last element in the sequence.）