如何为结构体中的Vector正确实现AddAssign trait？

你的代码存在两个核心问题：

1. 引用权限不匹配：你为&ColumnVector（不可变引用类型）实现了AddAssign，此时add_assign方法的第一个参数&mut self实际是&mut &ColumnVector——也就是对不可变引用的可变引用。但不可变引用本身不允许修改原ColumnVector的内部数据data，所以编译时会报错无法修改self.data。
2. 不符合操作符重载常规逻辑：Rust中操作符重载通常针对自身类型（如ColumnVector）实现，而非引用类型，这种实现方式会导致调用时需要额外的引用嵌套，不符合开发者的使用习惯。

按照AddAssign trait的设计，你可以为ColumnVector实现AddAssign<&ColumnVector>（即指定右操作数类型为不可变引用），这样既不需要修改trait的方法签名，又能正确完成向量加法赋值操作：

use std::ops::AddAssign;

pub struct ColumnVector {
    pub data: Vec<f32>,
}

// 为ColumnVector实现AddAssign，右操作数为&ColumnVector
impl AddAssign<&ColumnVector> for ColumnVector {
    fn add_assign(&mut self, other: &ColumnVector) {
        // 先确保两个向量长度一致（可选，避免索引越界）
        assert_eq!(self.data.len(), other.data.len(), "Vectors must have the same length");
        
        for (index, elem) in other.data.iter().enumerate() {
            self.data[index] += elem;
        }
    }
}

fn main() {
    // 测试示例
    let mut vec1 = ColumnVector { data: vec![1.0, 2.0, 3.0] };
    let vec2 = ColumnVector { data: vec![4.0, 5.0, 6.0] };
    
    vec1 += &vec2;
    println!("{:?}", vec1.data); // 输出 [5.0, 7.0, 9.0]
}

方案可行性说明 #

AddAssign trait的定义是pub trait AddAssign<Rhs = Self> { fn add_assign(&mut self, rhs: Rhs); }，其中Rhs是关联类型，默认等于Self，但我们可以显式指定Rhs为&ColumnVector，完全符合trait的要求，不需要修改方法签名。此时&mut self是对ColumnVector的可变引用，拥有修改内部data的权限，other是不可变引用，符合只读访问的需求。

如果你坚持要为可变引用类型实现AddAssign，也可以这样写，但调用逻辑不够直观：

impl AddAssign<&ColumnVector> for &mut ColumnVector {
    fn add_assign(&mut self, other: &ColumnVector) {
        assert_eq!(self.data.len(), other.data.len());
        for (index, elem) in other.data.iter().enumerate() {
            self.data[index] += elem;
        }
    }
}

// 调用方式
let mut vec1 = ColumnVector { data: vec![1.0, 2.0] };
let vec2 = ColumnVector { data: vec![3.0, 4.0] };
(&mut vec1) += &vec2;

内容的提问来源于stack exchange，提问作者Brian Yeh