如何从二维Vec构造引用类型的一维Vec？

原代码的核心问题 #

1. 变量名不一致：定义的容器是grid，后续却使用了未定义的field，需统一变量名。
2. 迭代器使用错误：into_iter是方法，必须加括号调用()；且into_iter会转移原容器的所有权，若仅需元素引用而非消耗原网格，应使用iter()（遍历不可变引用）。
3. 语法错误：collect是方法需加括号调用，且无需在结果前加&——原写法错误地获取了收集结果的引用，导致类型不匹配。
4. 可变需求缺失：调用pop()需要Vec是可变的，需给cell_stack添加mut关键字。

修正后的代码示例 #

#[derive(Debug, PartialEq, Eq, PartialOrd, Ord)]
struct DataStruct(i32); // 示例结构体，实现排序所需的Trait

fn main() {
    let mut grid: Vec<Vec<DataStruct>> = Vec::new();
    // 初始化网格示例
    grid.push(vec![DataStruct(3), DataStruct(1)]);
    grid.push(vec![DataStruct(4), DataStruct(2)]);

    // 创建包含所有元素不可变引用的一维Vec
    let mut cell_stack: Vec<&DataStruct> = grid.iter()
        .flatten()
        .collect();

    // 按默认规则排序（可替换为sort_by自定义排序逻辑）
    cell_stack.sort();

    // 迭代处理元素
    while let Some(cell) = cell_stack.pop() {
        some_func(cell);
    }
}

fn some_func(cell: &DataStruct) {
    println!("处理元素: {:?}", cell);
}

关键细节说明 #

• 若需要可变引用（修改原网格元素），可将grid.iter()替换为grid.iter_mut()，收集为Vec<&mut DataStruct>。
• 若无需保留原网格，可使用into_iter()消耗原容器，收集元素所有权而非引用：

  let mut cell_stack: Vec<DataStruct> = grid.into_iter()
      .flatten()
      .collect();
  

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