Rust中已知长度Vec转数组的低开销高效方法探究

安全且零拷贝的稳定版方案 #

在Rust 1.48及以上版本中，Vec<T>实现了TryInto<[T; N]> trait，可直接将长度匹配的Vec转换为数组，全程无拷贝、无额外堆操作：

use std::convert::TryInto;

fn vec_to_array(vec: Vec<i32>) -> Result<[i32; 4], Vec<i32>> {
    vec.try_into()
}

当输入Vec的长度恰好为4时，该方法会直接接管Vec的堆内存，转换为数组返回；若长度不匹配，会将原Vec完整返回，方便处理错误分支。这个方案完全安全，性能拉满，是优先推荐的写法。

手动unsafe实现（需确保长度匹配） #

如果你的场景能100%保证Vec长度等于数组大小，也可以用unsafe代码手动完成内存转换，避免错误处理的开销：

fn vec_to_array(vec: Vec<i32>) -> [i32; 4] {
    // 必须确保长度匹配，否则会触发未定义行为
    assert_eq!(vec.len(), 4);
    
    // 拆解Vec的底层指针、长度、容量
    let (ptr, len, cap) = vec.into_raw_parts();
    debug_assert_eq!(len, 4);
    
    // 从指针读取整个数组
    let array = unsafe { std::ptr::read(ptr as *const [i32; 4]) };
    
    // 释放原Vec的堆内存，避免泄漏
    unsafe {
        std::alloc::dealloc(
            ptr as *mut u8,
            std::alloc::Layout::array::<i32>(cap).unwrap(),
        );
    }
    
    array
}

这个写法完全复用原Vec的内存，没有任何拷贝操作，但必须严格保证输入Vec的长度正确，否则会导致内存错误。

原有方案问题分析 #

你之前的代码只读取了指针指向的第一个元素，并没有构造完整数组，还会导致原Vec的堆内存泄漏，同时没有实现正确的数组转换逻辑，所以性能和正确性都有问题。

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