如何在没有临时设置的s1中存储设置差异s1-s2?

时间:2013-12-25 12:19:02

标签: c++ stl set

STL是否已包含任何简单的方法或算法,用于直接在set1中存储集set2set1之间的差异,而无需临时集变量?

下面的示例代码显示了我已尝试过的一些替代方案(哪些方法无效)以及带有临时集tmp的解决方案(我想避免):

int _tmain(int argc, _TCHAR* argv[])
{
    std::set<int> set1, set2;

    set1.insert(1); set1.insert(2); set1.insert(3); set1.insert(4); set1.insert(5);
    set2.insert(4); set2.insert(6);

    // NONE OF THE FOLLOWING ALTERNATIVES DID WORK:
    // a: // set1.erase(set2.end(), set2.begin());
    // b: // std::set_difference(set1.begin(), set1.end(), 
    //                set2.begin(), set2.end(), set1.begin());
    // c: // std::remove_if(set1.begin(), set1.end(), 
    //                [set2](int i){return set2.find(i) != set2.end();} );

    // Complicated version, for which I am trying to find something simpler:
    std::set<int> tmp;
    std::set_difference(set1.begin(), set1.end(), set2.begin(), set2.end(), std::inserter(tmp, tmp.end()));
    set1.clear();
    std::copy(tmp.begin(), tmp.end(), std::inserter(set1, set1.end()));

    // Print result: // Expect 1 2 3 5
    std::cout << "set1: ";
    for (auto it=set1.begin(); it != set1.end(); it++)
    {
        std::cout << *it << " ";
    }
    std::cout << std::endl;

    return 0;
}

我正在寻找一个需要C ++ 11的解决方案(少数C++11 constructs allowed in Visual Studio 2010除外)。

3 个答案:

答案 0 :(得分:2)

这似乎很容易手动编写:

std::set<int>::iterator iter1 = set1.begin();
std::set<int>::iterator end1 = set1.end();

std::set<int>::const_iterator iter2 = set2.begin();
std::set<int>::const_iterator end2 = set2.end();

while (iter1 != end1 && iter2 != end2) {
  if (*iter1 < *iter2) {
    ++iter1;
  } else if (*iter2 < *iter1) {
    ++iter2;
  } else {
    set1.erase(iter1++);
    ++iter2;
  }
}

您可以将其置于通用的可重用函数中。

答案 1 :(得分:1)

你可以通过简单的仿函数类来避免临时删除,如下所示:

class remover {
    std::set<int> &s;
public:
    remover(std::set<int>& theSet) : s(theSet) {}
    void operator()(int val) { s.erase(val); }
};

掌握此类后,您可以按如下方式对删除进行编码:

for_each(set2.begin(), set2.end(), remover(set1));

这个想法类似于基于lambdas的解决方案#c,来自已注释掉的无效工作列表。

Demo on ideone

答案 2 :(得分:0)

我正在寻找更好的解决方案,但作为一个开始,您可以通过一些修改来编写此内容,并删除对std::copyclear

的不必要使用
#include <iostream>
#include <set>
#include <algorithm>

int main(int argc, char *argv[]) {
  std::set<int> set1, set2;

  set1.insert(1);
  set1.insert(2);
  set1.insert(3);
  set1.insert(4);
  set1.insert(5);

  set2.insert(4);
  set2.insert(6);
  {
    std::set<int> tmp;
    std::set_difference(set1.begin(), set1.end(), set2.begin(), set2.end(),
                        std::inserter(tmp, tmp.end()));

    set1 = std::set<int>(tmp.begin(), tmp.end());
  }
  std::cout << "set1: ";
  for (auto it = set1.begin(); it != set1.end(); it++) {
    std::cout << *it << " ";
  }
  std::cout << std::endl;

  return 0;
}

范围也等于C ++中的“免费RAII”......