algorithm - Rust 有没有计算乘积之和的好方法？

Question

我是 Rust 新手，我正在寻找一种计算 Rust 中乘积之和的好方法。

我的意思是如果有一个像[3,4,5]这样的向量， 我想计算 (3 * 4) + (3 * 5) + (4 * 5)。

这是我的方法，

fn main() {
    let a = [3, 4, 5];
    assert_eq!(
        47,
        [3, 4, 5].iter().enumerate().fold(0, |mut s, (i, v)| {
            for ii in i + 1..a.len() {
                s += *v * a[ii];
            }
            s
        })
    );
}

如果您知道更好的，我很想知道!

Answer 1

您可以使用组合、map和sum或组合和fold实现产品。

use itertools::Itertools;
fn main() {
    let a = [3, 4, 5];

    let b: i32 = a.iter().combinations(2).map(|x| x[0] * x[1]).sum();
    assert_eq!(b, 47);

    let c: i32 = a.iter().combinations(2).fold(0, |acc, x| acc + (x[0] * x[1]) );
    assert_eq!(c, 47);

}

playground