有没有一种方法可以创建一个闭包来计算Rust中具有随机系数的n次多项式

发布于09月21日

所以我们的 idea 是有一个函数，它返回带有随机系数的n次闭包.例如，这对于比较根查找方法可能很有用，因为我们可以自动创建要计算的函数. 将任何操作附加到闭包的返回将是一个更强大的资源，所以如果有什么方法可以做到这一点，请告诉我.

代码应如下所示:

use rand::Rng;
fn create_func(n: u8) -> impl Fn(f64) -> f64 { 
    let mut cst = vec![];
    for _ in 0..n{
        cst.push(rand::thread_rng().gen_range(0..=20)  as f64);
        // establish the coefficients by randomly generating them
    }
    move |x| cst[0]*x + cst[1]*x.powf(1.) + /*...*/ cst[n- 1]*x.powf(n as f64)
}

所以很明显，最后一部分并不是表达我们want to返回的闭包的其余部分的有效方式，但我希望您能理解.

推荐答案

您正在寻找和，因此将您的序列转换为将被求和的迭代器.

move |x| {
    cst.iter()
        .zip(0..=n as i32)
        .map(|(&c, exp)| c * x.powi(exp))
        .sum()
}

对于每一项，您需要cst(iter)及其指数(zip)中的元素.您可以从这些(map)中一次构造一个术语，然后将所有术语相加(sum).

我还固定了你的词条数量(你的没有x^0个词条)，使用了sample_iter个，这比每次都呼叫thread_rng要快.我换成了powi，因为所有的指数都是整数.

pub fn create_func(n: u8) -> impl Fn(f64) -> f64 {
    let cst: Vec<f64> = rand::thread_rng()
        .sample_iter(rand::distributions::Uniform::new_inclusive(0, 20))
        .map(|n| n as f64)
        .take(n as usize + 1)
        .collect();

    move |x| {
        cst.iter()
            .zip(0..=n as i32)
            .map(|(&c, exp)| c * x.powi(exp))
            .sum()
    }
}