如何在Rust的Itertools中实现可重复排列（Dispositions with repetition）？

需求回顾 #

• 生成的序列顺序有意义（不同顺序视为不同结果）
• 允许从原集合中重复选取元素
• 生成的序列长度固定为指定值k

以元素集合[0,1]、k=3为例，预期生成所有3位二进制序列：

[0, 0, 0]
[0, 0, 1]
[0, 1, 0]
[0, 1, 1]
[1, 0, 0]
[1, 0, 1]
[1, 1, 0]
[1, 1, 1]

实现方案 #

Itertools库中没有直接对应"可重复排列"的命名API，但这类需求本质是集合自身的k次笛卡尔积，可以通过repeat结合multi_cartesian_product方法实现，该方案支持动态指定k值。

步骤1：添加依赖 #

在Cargo.toml中引入Itertools：

[dependencies]
itertools = "0.12" # 可替换为最新版本

步骤2：代码实现 #

use itertools::Itertools;

fn main() {
    let elements = [0, 1];
    let k = 3;

    // 生成k次笛卡尔积，即满足需求的可重复排列
    let sequences = elements.iter().cloned().repeat(k).multi_cartesian_product();

    // 遍历并打印所有结果
    for seq in sequences {
        println!("{:?}", seq);
    }
}

代码说明 #

• elements.iter().cloned()：将原集合转为可复制的迭代器，避免处理引用
• repeat(k)：将原迭代器重复k次，为笛卡尔积提供k个独立的元素选择源
• multi_cartesian_product()：计算这些迭代器的笛卡尔积，输出所有长度为k的有序元素组合（允许重复选取）

替代方案（固定k值场景） #

如果k是固定常量，也可以使用iproduct!宏直接指定k个集合迭代器：

use itertools::iproduct;

fn main() {
    let elements = [0, 1];
    
    for (a, b, c) in iproduct!(&elements, &elements, &elements) {
        println!("{:?}", [a, b, c]);
    }
}

内容的提问来源于stack exchange，提问作者Umberto Fontanazza