如何在 C++ 中两两访问 std::set 中的元素

Question

我有一个整数列表。（当前存储在 std::vector 中，但为了提高效率，我需要将其转换为 set。但在当前版本中，我使用它如下：（我使用的是 c+ +98 不是 c++11)

int res=0;
vector<vector<int> >costMatrix;
vector<int>partialSolution;
    for(int i =0;i<partialSolution.size()-1;i++){
        res+=costMatrix[partialSolution.get(i)][partialSolution.get(i+1)];
    }

所以，我需要对设置的数据结构做同样的事情。但我不知道如何一次从集合中获取两个元素。我可以使用下面的代码获得partialSolution.get(i)，但我无法获得partialSolution.get(i+1)。有人帮我修改下面的代码吗？

 // this time set<int> partialSolution
    int res=0;
    std::set<int>::iterator it;
    for (it = partialSolution.begin(); it != partialSolution.end(); ++it)
{
    res+=costMatrix[*it][]; 
}

Answer 1

这可以工作（从begin() 迭代到end()-1 并使用std::next 或++ 来获取当前项目旁边的项目）。

在 C++11 中：

for (it = partialSolution.begin(); it != std::prev(partialSolution.end()); ++it)
{
    res+=costMatrix[*it][*(std::next(it))]; 
}

在 C++98 中：

std::set<int>::iterator last = partialSolution.end();
--last;
for (it = partialSolution.begin(); it != last; ++it)
{
    // not optimal but I'm trying to make it easy to understand...
    std::set<int>::iterator next = it;
    ++next;
    res+=costMatrix[*it][*next]; 
}