问题描述
我正在向 RosettaCode 贡献 Rust 代码,以便同时学习 Rust 并为 Rust 社区做出贡献.在可变 Vec
中弹出最后一个 n 元素的最佳惯用方法是什么?
I am contributing Rust code to RosettaCode to both learn Rust and contribute to the Rust community at the same time. What is the best idiomatic way to pop the last n elements in a mutable Vec
?
以下是我写的大致内容,但我想看看是否有更好的方法:
Here's roughly what I have written but I'm wanting to see if there's a better way:
fn main() {
let mut nums: Vec<u32> = Vec::new();
nums.push(1);
nums.push(2);
nums.push(3);
nums.push(4);
nums.push(5);
let n = 2;
for _ in 0..n {
nums.pop();
}
for e in nums {
println!("{}", e)
}
}
(游乐场链接)
推荐答案
我建议使用 Vec::truncate
:
I'd recommend using Vec::truncate
:
fn main() {
let mut nums = vec![1, 2, 3, 4, 5];
let n = 2;
let final_length = nums.len().saturating_sub(n);
nums.truncate(final_length);
println!("{:?}", nums);
}
另外,我
- 使用了
saturating_sub
处理向量中没有N
元素的情况 - 使用了
vec![]
轻松构建数字向量 - 一次性打印出整个向量
通常,当您弹出"某些东西时,您希望拥有这些值.如果你想要另一个向量中的值,你可以使用 Vec::split_off:
Normally when you "pop" something, you want to have those values. If you want the values in another vector, you can use Vec::split_off
:
let tail = nums.split_off(final_length);
如果你想访问元素但不想创建一个全新的向量,你可以使用 Vec::drain
:
If you want access to the elements but do not want to create a whole new vector, you can use Vec::drain
:
for i in nums.drain(final_length..) {
println!("{}", i)
}
这篇关于在可变 Vec 中弹出最后 N 个元素的惯用方法是什么?的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持!