Rust 函数的向量返回类型错误

Question

pub fn collect_prime_factors(number: i32) -> Vec<i32> {
    let mut prime_factors = Vec::new();

    for i in 2..number {
        if number % i == 0 {
            collect_prime_factors(number / 2);
            prime_factors.push(i);
            prime_factors
        }
    }
}

错误：

lib.rs:14:9: 14:22 error: mismatched types:
 expected `()`,
    found `collections::vec::Vec<i32>`
(expected (),
    found struct `collections::vec::Vec`) [E0308]
lib.rs:14         prime_factors

我不明白这里的问题。我将 Vec<i32> 声明为返回类型。为什么期待那些空括号？

为什么只有在循环中使用它时这不起作用？当我删除循环并且只有return prime_factors; 一切正常。

Answer 1

有两个问题（除了粘贴错误）。

你引用的错误不是函数的返回值；这是if 表达式的值：

pub fn collect_prime_factors(number: i32) -> Vec<i32> {
    let mut prime_factors = Vec::new();

    for i in 2..number {
        if number % i == 0 {
            prime_factors.push(i);
            prime_factors   // This would be the value of the if
        }
    }
}

Rust 期望没有返回值，或者() 的值，但您返回的是prime_factors。

如果你解决了这个问题，你会看到下一个错误是相反的，它期望函数返回一个Vec<i32>，但你返回的是()（什么都没有）。

我认为这里正确的做法是在收集完所有因素后在函数末尾返回向量：

pub fn collect_prime_factors(number: i32) -> Vec<i32> {
    let mut prime_factors = Vec::new();

    for i in 2..number {
        if number % i == 0 {
            prime_factors.push(i);
        };
    }
    prime_factors  // Return the vector from the function.
}

Playground link

（但这个函数实际上并不只返回素数！）