如何迭代特定范围的 std::set/std::multiset？

Question

对于像 std::set<std::string> set={ "aaa", "bbb", "ccc", "ddd", "eee", "fff" }; 这样的 std::set 如何在不跟踪额外索引的情况下迭代该集合的特定范围 [a, b]？

类似这样的：

for(auto it = set.begin(); it!=set.begin()+b; ++it)
   std::cout << *it << " ";

或者像这样：

for(auto it = set.begin()+a; it!=set.begin()+b; ++it)
   std::cout << *it << " ";

a<=b 和 b<=set.size() 的位置

Answer 1

你不能做set.begin()+a，但你可以做std::advance(it, a)：

#include <iostream>
#include <set>
#include <string>
 
int main()
{
    std::set<std::string> set={ "aaa", "bbb", "ccc", "ddd", "eee", "fff" };
    size_t a = 1, b = 4;
    auto it = set.begin(), it_end = set.begin();
    std::advance(it, a);
    std::advance(it_end, b);
    for(; it!= it_end; ++it)
        std::cout << *it << " ";
}

很遗憾，您不能执行auto it = std::advance(set.begin(), a); 之类的操作，因为advance 通过引用获取迭代器并就地更改它。

https://ideone.com/8nRjgr

更好的解决方案使用std::next，感谢@Evg：

#include <iostream>
#include <set>
#include <string>
 
int main()
{
    std::set<std::string> set={ "aaa", "bbb", "ccc", "ddd", "eee", "fff" };
    size_t a = 1, b = 4;
    for(auto it = std::next(set.begin(), a), it_end = std::next(it, b-a); it != it_end; ++it)
        std::cout << *it << " ";
}

https://ideone.com/6wgpMZ

Answer 2

你可以编写自己的slice 类，有点像std::span：

#include <cstdio>
#include <set>
#include <string>

template <class Iter>
class slice {
 private:
  Iter _beg;
  Iter _end;

 public:
  slice(Iter beg, Iter end) noexcept 
    : _beg{beg}
    , _end{end} 
  {}

  slice(Iter beg, std::size_t sz) noexcept 
    : slice(beg, std::next(beg, sz)) 
  {}

  auto begin() const noexcept { return _beg; }
  auto end() const noexcept { return _end; }
};

int main() {
  std::set<std::string> const set{"aaa", "bbb", "ccc", "ddd", "eee", "fff"};

  slice const sl(set.begin(), 3);

  for (auto const& elm : sl) std::puts(elm.c_str());
}

Godbolt