在不改变向量的情况下返回 Rust 中向量的最后 n 个元素

发布于06月15日

我正在努力寻找一种方法，从一个向量的末端取两个值，求和这些值，然后把和推到向量上.

我发现pop、truncate和drain不起作用，因为它们从原始向量中移除了值.

fn main() {
    println!("Which Fibonacci number would you like to find?");

    let mut fib_num = String::new();

    io::stdin().read_line(&mut fib_num)
        .expect("Failed to read line");

    let fib_num: u32 = fib_num.trim().parse()
        .expect("Please enter a number");

    let mut stored_nums: Vec<u32> = vec![0, 1];

    while fib_num > stored_nums.len() as u32 {
        let mut limit = stored_nums.len();
        let mut new_num1 = stored_nums.pop().unwrap();
        let mut new_num2 = stored_nums.pop().unwrap_or(0);
        stored_nums.push(new_num1 + new_num2);
    }
}

推荐答案

你需要考虑向量doesn't have two items的情况.

我会使用迭代器适配器，比如Iterator::rev和Iterator::take，最后使用Iterator::sum:

let sum = stored_nums.iter().rev().take(2).sum();
stored_nums.push(sum);

这允许您避免explicit次处理向量/切片/迭代器太短但代码仍隐式处理它的情况.

您还可以直接索引到切片中:

let len = stored_nums.len();
let sum = stored_nums[len - 1] + stored_nums[len - 2];
stored_nums.push(sum);

然而，如果少于2个元素，这将导致panic .

在这种情况下，您可以try 处理向量太短的问题，但它有点冗长:

fn add_last_two(nums: &[u32]) -> Option<u32> {
    let len = nums.len();

    let idx_a = len.checked_sub(1)?;
    let idx_b = len.checked_sub(2)?;

    let a = nums.get(idx_a)?;
    let b = nums.get(idx_b)?;

    Some(a + b)
}

fn main() {
    let mut stored_nums: Vec<u32> = vec![0, 1];
    let sum = add_last_two(&stored_nums).unwrap_or(0);
    stored_nums.push(sum);
}

请注意，在Vec中使用a Fibonacci iterator和collect可能更好.