c++ - 如何组合范围 v3 变换、连接和循环？

Question

我正在尝试使用 range-v3 库生成无限范围的整数。我的目标是 {0,0,0,1,1,1,2,2,2,0,0,0,1,1,1 ...}。我希望每个数字重复的次数也是可变的。没有这个限制，我可以做到这一点，而且效果很好。

const std::vector<std::vector<int>> repvec = { {0,0,0},{1,1,1},{2,2,2} };
auto joinedVec = repvec | ranges::view::join;
auto infiniteVec = joinedVec | ranges::view::cycle;

但是如果我想要可变的重复次数，我就无法编译它:

int nbRep = 3;
const std::vector<int> origVec = { 0, 1, 2 };
auto repVec = origVec | ranges::view::transform( [nbRep] (const int &x) { 
    return ranges::view::repeat(x) | ranges::views::take(nbRep); 
});
auto joinedVec = repVec | ranges::view::join;
auto infiniteVec = joinedVec | ranges::view::cycle;

我不确定这是否是范围库的限制或者我是否做错了什么。

Answer 1

我真的不明白你为什么要这样做:

ranges::view::repeat(x) | ranges::views::take(nbRep)

而不仅仅是:

ranges::view::repeat_n(nbRep)

解决方案非常简单:

auto generate_range(const std::vector<int>& original, const std::size_t reps) {
    using namespace ranges; 

    auto rng = original | views::transform([reps](const auto arg) {
        return views::repeat_n(arg, reps);
    });

    return rng | views::cycle | views::join;
}

我们首先根据原始 vector 创建一个范围的范围 - 我们将其元素转换为范围的重复元素。我们最终得到一个范围，其中的元素是由重复值组成的范围。根据您的示例，我们有:

rng = [[0, 0, 0], [1, 1, 1], [2, 2, 2]]

现在我们只需要循环所有元素即可。我们最终得到:

rng | cycle = [[0, 0, 0], [1, 1, 1], [2, 2, 2], [0, 0, 0], [1, 1, 1], ...]

我们需要做的最后一件事是加入他们:

rng | cycle | join = [0, 0, 0, 1, 1, 1, 2, 2, 2, 0, 0, 0, 1, 1, 1, ...]

完整示例:

#include <iostream>
#include <range/v3/all.hpp>
#include <vector>

auto generate_range(const std::vector<int>& original, const std::size_t reps) {
    using namespace ranges;

    auto rng = original | views::transform([reps](const auto arg) {
        return views::repeat_n(arg, reps);
    });

    return rng | views::cycle | views::join;
}

int main() {
    using namespace ranges;

    const std::vector<int> original = {0, 1, 2};
    const int reps = 3;

    for (const auto r : generate_range(original, reps) | ranges::views::take(15)) {
        std::cout << r << ' ';
    }
}

打印:0 0 0 1 1 1 2 2 2 0 0 0 1 1 1。