Rust中收集固定数量元素时如何设置Vec的初始容量?
解决方案
你原来的写法错误在于混淆了类型标注和实例创建的语法,Vec<i32>::with_capacity(2)是创建Vec实例的表达式,不能放在变量的类型位置。以下是几种正确实现预分配容量的方式:
方法1:先创建带容量的Vec,再用extend填充
// 先初始化容量为2的Vec let mut info = Vec::with_capacity(2); // 分割字符串、解析为i32后填充到Vec中 // 推荐用split_whitespace替代split(" "),避免连续空格产生空字符串元素 info.extend(buffer.trim().split_whitespace().map(|s| s.parse().unwrap()));
方法2:利用collect的容器复用特性
let mut info = Vec::with_capacity(2); // collect会尝试复用已有Vec的容量,这里因为已知元素数量是2,不会触发重新分配 info = buffer.trim().split_whitespace() .map(|s| s.parse().unwrap()) .collect();
方法3:使用collect_into(Rust 1.55+)
这个方法更直观,直接将迭代器元素收集到预先分配好的Vec中:
use std::iter::Iterator; let mut info = Vec::with_capacity(2); buffer.trim().split_whitespace() .map(|s| s.parse().unwrap()) .collect_into(&mut info);
额外建议:确保元素数量符合预期
既然你明确知道应该获取到2个元素,推荐使用数组来强制校验数量,再转成Vec(Rust 1.63+支持try_collect):
use std::array; // 尝试收集为固定长度数组,元素数量不符或解析失败时会触发panic(可根据需求替换为match处理) let info_array: [i32; 2] = buffer.trim().split_whitespace() .map(|s| s.parse()) .try_collect() .expect("必须解析出恰好2个整数"); // 转成Vec,容量自动为2 let mut info = info_array.to_vec();
内容的提问来源于stack exchange,提问作者Pioneer_11
相关产品推荐
相关产品推荐

