如何结合范围 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;
}

我们首先基于 original 向量创建一个 range of ranges - 我们 transform 其元素为 ranges 重复元素。我们最终得到一个范围，其中元素是由重复值组成的范围。根据您的示例，我们有：

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], ...]

我们需要做的最后一件事是 join 他们：

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.

如何结合范围 v3 转换、连接和循环？

how to combine ranges v3 transform, join and cycle?

c++

range-v3

c++17