从 C++ 走向 rust（十）：泛型算法——向量加法

最新推荐文章于 2022-10-18 16:56:26 发布

原创最新推荐文章于 2022-10-18 16:56:26 发布 · 620 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#rust #泛型 #向量 #加法

Rust 同时被 2 个专栏收录

233 篇文章

订阅专栏

程序设计语言

32 篇文章

订阅专栏

博主分享了使用Rust语言编写科学计算库的经历，从向量加法运算开始，实现了RowVec结构体的Add trait。代码涉及变量引用、解引用，并通过测试确保功能正确。Rust的内存安全特性给博主留下了深刻印象。

打算写一个科学计算库，从向量运算入手。费了一天功夫，终于完成了这十来行代码。这里面牵涉到变量引用、解引用，也不知道用法是否正确。感觉 rust 从各方面颠覆我的想象。

use std::ops::{Add, Sub, Mul, Div};

pub struct RowVec<T>(Vec<T>);
impl <T: Add<Output = T> + Copy + Clone> Add for RowVec<T> {
    type Output = Self;
    fn add(self, other: Self) -> Self {
         let x = self.0.iter();
         let y = other.0.iter();
         let z = x.zip(y).map(|(a,b)| *a + *b).collect::<Vec<T>>();
         Self(z)
    }
}
#[test]
fn test_add() {
    let x = RowVec(vec![1,2,3]);
    let y = RowVec(vec![4,5,6]);
    let z = x + y;
    for p in z.0.iter() {
        println!("{:?}", p);
    }
}