如何在Rust中获取固定大小的所有元素组合？

如果你需要的是所有长度为k的元素组合（元素不重复、保持原Vec中的顺序），而非连续片段，Rust标准库没有直接提供对应工具，但可以通过两种方式实现：

方法一：使用itertools crate（推荐） #

itertools是Rust生态中常用的迭代器工具库，其中的combinations方法可直接生成所需组合。

1. 在Cargo.toml中添加依赖：

[dependencies]
itertools = "0.12"

2. 代码示例：

use itertools::Itertools;

fn main() {
    let nums = vec![1, 2, 3, 4, 5, 6];
    // 生成所有长度为4的组合
    let combinations: Vec<Vec<i32>> = nums.into_iter().combinations(4).collect();
    
    // 打印所有组合
    for combo in combinations {
        println!("{:?}", combo);
    }
}

这段代码会输出全部C(6,4)=15种组合，包含你提到的[1,2,3,4]、[1,3,4,5]等结果。

方法二：手动实现组合生成（无需第三方库） #

如果不想依赖外部库，可以通过回溯法生成合法索引组合，再提取对应元素：

fn combinations<T: Clone>(nums: &[T], k: usize) -> Vec<Vec<T>> {
    let mut result = Vec::new();
    let mut current = Vec::new();
    
    fn backtrack<T: Clone>(nums: &[T], k: usize, start: usize, current: &mut Vec<T>, result: &mut Vec<Vec<T>>) {
        if current.len() == k {
            result.push(current.clone());
            return;
        }
        
        for i in start..nums.len() {
            current.push(nums[i].clone());
            backtrack(nums, k, i + 1, current, result);
            current.pop();
        }
    }
    
    backtrack(nums, k, 0, &mut current, &mut result);
    result
}

fn main() {
    let nums = vec![1, 2, 3, 4, 5, 6];
    let combos = combinations(&nums, 4);
    
    for combo in combos {
        println!("{:?}", combo);
    }
}

该实现通过回溯遍历所有合法元素选择，保证组合元素的位置和顺序与原Vec一致。

内容的提问来源于stack exchange，提问作者Juan Marchetto