AI 助理
备案控制台
开发者社区 开发与运维 文章 正文

Rust 重载运算符｜复数结构的“加减乘除”四则运算

2024-01-30 104
版权
版权声明：
本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《 阿里云开发者社区用户服务协议》和 《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写 侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。
简介： Rust 重载运算符｜复数结构的“加减乘除”四则运算


复数

基本概念

复数定义

由实数部分和虚数部分所组成的数，形如a＋bi 。

其中a、b为实数，i 为“虚数单位”，i² = -1，即虚数单位的平方等于-1。

a、b分别叫做复数a＋bi的实部和虚部。

当b=0时，a＋bi＝a 为实数；

当b≠0时，a＋bi 又称虚数；

当b≠0、a=0时，bi 称为纯虚数。

实数和虚数都是复数的子集。如同实数可以在数轴上表示一样复数也可以在平面上表示，复数x+yi以坐标点(x，y)来表示。表示复数的平面称为“复平面”。

复数相等

两个复数不能比较大小，但当个两个复数的实部和虚部分别相等时，即表示两个复数相等。

共轭复数

如果两个复数的实部相等，虚部互为相反数，那么这两个复数互为共轭复数。

复数的模

复数的实部与虚部的平方和的正的平方根的值称为该复数的模，数学上用与绝对值“|z|”相同的符号来表示。虽然从定义上是不相同的，但两者的物理意思都表示“到原点的距离”。

复数的四则运算

加法（减法）法则

复数的加法法则：设z1=a+bi,z2 =c+di是任意两个复数。两者和的实部是原来两个复数实部的和，它的虚部是原来两个虚部的和。两个复数的和依然是复数。

即(a+bi)±(c+di)=(a±c)+(b±d)

乘法法则

复数的乘法法则：把两个复数相乘，类似两个多项式相乘，结果中i²=-1，把实部与虚部分别合并。两个复数的积仍然是一个复数。

即(a+bi)(c+di)=(ac-bd)+(bc+ad)i

除法法则

复数除法法则：满足(c+di)(x+yi)=(a+bi)的复数x+yi(x,y∈R)叫复数a+bi除以复数c+di的商。

运算方法：可以把除法换算成乘法做，将分子分母同时乘上分母的共轭复数，再用乘法运算。

即(a+bi)/(c+di)=(a+bi)(c-di)/(c*c+d*d)=[(ac+bd)+(bc-ad)i]/(c*c+d*d)

复数的Rust代码实现

结构定义

Rust语言中，没有像python一样内置complex复数数据类型，我们可以用两个浮点数分别表示复数的实部和虚部，自定义一个结构数据类型，表示如下：

struct Complex {
    real: f64,
    imag: f64,
}

示例代码：

#[derive(Debug)]
struct Complex {
    real: f64,
    imag: f64,
}
impl Complex {  
    fn new(real: f64, imag: f64) -> Self {
        Complex { real, imag }  
    }
}
fn main() {  
    let z = Complex::new(3.0, 4.0);
    println!("{:?}", z);
    println!("{} + {}i", z.real, z.imag);
}

注意：#[derive(Debug)] 自动定义了复数结构的输出格式，如以上代码输出如下：

Complex { real: 3.0, imag: 4.0 }

3 + 4i

重载四则运算

复数数据结构不能直接用加减乘除来做复数运算，需要导入标准库ops的运算符：

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

Add, Sub, Mul, Div, Neg 分别表示加减乘除以及相反数，类似C++或者python语言中“重载运算符”的概念。

根据复数的运算法则，写出对应代码：

fn add(self, other: Complex) -> Complex {
    Complex {
        real: self.real + other.real,
        imag: self.imag + other.imag,
    }  
}  
fn sub(self, other: Complex) -> Complex {
    Complex {  
        real: self.real - other.real,
        imag: self.imag - other.imag,
    }  
} 
fn mul(self, other: Complex) -> Complex {  
    let real = self.real * other.real - self.imag * other.imag;
    let imag = self.real * other.imag + self.imag * other.real;
    Complex { real, imag }  
}  
fn div(self, other: Complex) -> Complex {
    let real = (self.real * other.real + self.imag * other.imag) / (other.real * other.real + other.imag * other.imag);
    let imag = (self.imag * other.real - self.real * other.imag) / (other.real * other.real + other.imag * other.imag);
    Complex { real, imag }
}
fn neg(self) -> Complex {
    Complex {
        real: -self.real,
        imag: -self.imag,
    }
}

Rust 重载运算的格式，请见如下示例代码：

use std::ops::{Add, Sub, Mul, Div, Neg};
#[derive(Clone, Debug, PartialEq)]
struct Complex {
    real: f64,
    imag: f64,
}
impl Complex {  
    fn new(real: f64, imag: f64) -> Self {
        Complex { real, imag }  
    }
    fn conj(&self) -> Self {
        Complex { real: self.real, imag: -self.imag }
    }
    fn abs(&self) -> f64 {
        (self.real * self.real + self.imag * self.imag).sqrt()
    }
}
fn abs(z: Complex) -> f64 {
    (z.real * z.real + z.imag * z.imag).sqrt()
}
impl Add<Complex> for Complex {
    type Output = Complex;
    fn add(self, other: Complex) -> Complex {
        Complex {
            real: self.real + other.real,
            imag: self.imag + other.imag,
        }  
    }  
}  
impl Sub<Complex> for Complex {
    type Output = Complex;
    fn sub(self, other: Complex) -> Complex {
        Complex {  
            real: self.real - other.real,
            imag: self.imag - other.imag,
        }  
    } 
}  
impl Mul<Complex> for Complex {
    type Output = Complex;  
    fn mul(self, other: Complex) -> Complex {  
        let real = self.real * other.real - self.imag * other.imag;
        let imag = self.real * other.imag + self.imag * other.real;
        Complex { real, imag }  
    }  
}
impl Div<Complex> for Complex {
    type Output = Complex;
    fn div(self, other: Complex) -> Complex {
        let real = (self.real * other.real + self.imag * other.imag) / (other.real * other.real + other.imag * other.imag);
        let imag = (self.imag * other.real - self.real * other.imag) / (other.real * other.real + other.imag * other.imag);
        Complex { real, imag }
    }
}  
impl Neg for Complex {
    type Output = Complex;
    fn neg(self) -> Complex {
        Complex {
            real: -self.real,
            imag: -self.imag,
        }
    }
}
fn main() {  
    let z1 = Complex::new(2.0, 3.0);
    let z2 = Complex::new(3.0, 4.0);
    let z3 = Complex::new(3.0, -4.0);
    // 复数的四则运算
    let complex_add = z1.clone() + z2.clone();
    println!("{:?} + {:?} = {:?}", z1, z2, complex_add);
    let complex_sub = z1.clone() - z2.clone();
    println!("{:?} - {:?} = {:?}", z1, z2, complex_sub);
    let complex_mul = z1.clone() * z2.clone();
    println!("{:?} * {:?} = {:?}", z1, z2, complex_mul);
    let complex_div = z2.clone() / z3.clone();
    println!("{:?} / {:?} = {:?}", z1, z2, complex_div);
    // 对比两个复数是否相等
    println!("{:?}", z1 == z2);
    // 共轭复数
    println!("{:?}", z2 == z3.conj());
    // 复数的相反数
    println!("{:?}", z2 == -z3.clone() + Complex::new(6.0,0.0));
    // 复数的模
    println!("{}", z1.abs());
    println!("{}", z2.abs());
    println!("{}", abs(z3));
}

输出：

Complex { real: 2.0, imag: 3.0 } + Complex { real: 3.0, imag: 4.0 } = Complex { real: 5.0, imag: 7.0 }

Complex { real: 2.0, imag: 3.0 } - Complex { real: 3.0, imag: 4.0 } = Complex { real: -1.0, imag: -1.0 }

Complex { real: 2.0, imag: 3.0 } * Complex { real: 3.0, imag: 4.0 } = Complex { real: -6.0, imag: 17.0 }

Complex { real: 2.0, imag: 3.0 } / Complex { real: 3.0, imag: 4.0 } = Complex { real: -0.28, imag: 0.96 }

false

true

true

3.605551275463989

5

5

示例代码中，同时还定义了复数的模 abs()，共轭复数 conj()。

两个复数的相等比较 z1 == z2，需要 #[derive(PartialEq)] 支持。

自定义 trait Display

复数结构的原始 Debug trait 表达的输出格式比较繁复，如：

Complex { real: 2.0, imag: 3.0 } + Complex { real: 3.0, imag: 4.0 } = Complex { real: 5.0, imag: 7.0 }

想要输出和数学中相同的表达(如 a + bi)，需要自定义一个 Display trait，代码如下：

impl std::fmt::Display for Complex {
    fn fmt(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
        if self.imag == 0.0 {
            formatter.write_str(&format!("{}", self.real))
        } else {
            let (abs, sign) = if self.imag > 0.0 {  
                (self.imag, "+" )
            } else {
                (-self.imag, "-" )
            };
            if abs == 1.0 {
                formatter.write_str(&format!("({} {} i)", self.real, sign))
            } else {
                formatter.write_str(&format!("({} {} {}i)", self.real, sign, abs))
            }
        }
    }
}

输出格式分三种情况：虚部为0，正数和负数。另外当虚部绝对值为1时省略1仅输出i虚数单位。

完整代码如下：

use std::ops::{Add, Sub, Mul, Div, Neg};
#[derive(Clone, PartialEq)]
struct Complex {
    real: f64,
    imag: f64,
}
impl std::fmt::Display for Complex {
    fn fmt(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
        if self.imag == 0.0 {
            formatter.write_str(&format!("{}", self.real))
        } else {
            let (abs, sign) = if self.imag > 0.0 {  
                (self.imag, "+" )
            } else {
                (-self.imag, "-" )
            };
            if abs == 1.0 {
                formatter.write_str(&format!("({} {} i)", self.real, sign))
            } else {
                formatter.write_str(&format!("({} {} {}i)", self.real, sign, abs))
            }
        }
    }
}
impl Complex {  
    fn new(real: f64, imag: f64) -> Self {
        Complex { real, imag }  
    }
    fn conj(&self) -> Self {
        Complex { real: self.real, imag: -self.imag }
    }
    fn abs(&self) -> f64 {
        (self.real * self.real + self.imag * self.imag).sqrt()
    }
}
fn abs(z: Complex) -> f64 {
    (z.real * z.real + z.imag * z.imag).sqrt()
}
impl Add<Complex> for Complex {
    type Output = Complex;
    fn add(self, other: Complex) -> Complex {
        Complex {
            real: self.real + other.real,
            imag: self.imag + other.imag,
        }  
    }  
}  
impl Sub<Complex> for Complex {
    type Output = Complex;
    fn sub(self, other: Complex) -> Complex {
        Complex {  
            real: self.real - other.real,
            imag: self.imag - other.imag,
        }  
    } 
}  
impl Mul<Complex> for Complex {
    type Output = Complex;  
    fn mul(self, other: Complex) -> Complex {  
        let real = self.real * other.real - self.imag * other.imag;
        let imag = self.real * other.imag + self.imag * other.real;
        Complex { real, imag }  
    }  
}
impl Div<Complex> for Complex {
    type Output = Complex;
    fn div(self, other: Complex) -> Complex {
        let real = (self.real * other.real + self.imag * other.imag) / (other.real * other.real + other.imag * other.imag);
        let imag = (self.imag * other.real - self.real * other.imag) / (other.real * other.real + other.imag * other.imag);
        Complex { real, imag }
    }
}  
impl Neg for Complex {
    type Output = Complex;
    fn neg(self) -> Complex {
        Complex {
            real: -self.real,
            imag: -self.imag,
        }
    }
}
fn main() {
    let z1 = Complex::new(2.0, 3.0);
    let z2 = Complex::new(3.0, 4.0);
    let z3 = Complex::new(3.0, -4.0);
    // 复数的四则运算
    let complex_add = z1.clone() + z2.clone();
    println!("{} + {} = {}", z1, z2, complex_add);
    let z = Complex::new(1.5, 0.5);
    println!("{} + {} = {}", z, z, z.clone() + z.clone());
    let complex_sub = z1.clone() - z2.clone();
    println!("{} - {} = {}", z1, z2, complex_sub);
    let complex_sub = z1.clone() - z1.clone();
    println!("{} - {} = {}", z1, z1, complex_sub);
    let complex_mul = z1.clone() * z2.clone();
    println!("{} * {} = {}", z1, z2, complex_mul);
    let complex_mul = z2.clone() * z3.clone();
    println!("{} * {} = {}", z2, z3, complex_mul);
    let complex_div = z2.clone() / z3.clone();
    println!("{} / {} = {}", z1, z2, complex_div);
    let complex_div = Complex::new(1.0,0.0) / z2.clone();
    println!("1 / {} = {}", z2, complex_div);
    // 对比两个复数是否相等
    println!("{:?}", z1 == z2);
    // 共轭复数
    println!("{:?}", z2 == z3.conj());
    // 复数的相反数
    println!("{:?}", z2 == -z3.clone() + Complex::new(6.0,0.0));
    // 复数的模
    println!("{}", z1.abs());
    println!("{}", z2.abs());
    println!("{}", abs(z3));
}

输出：

(2 + 3i) + (3 + 4i) = (5 + 7i)

(1.5 + 0.5i) + (1.5 + 0.5i) = (3 + i)

(2 + 3i) - (3 + 4i) = (-1 - i)

(2 + 3i) - (2 + 3i) = 0

(2 + 3i) * (3 + 4i) = (-6 + 17i)

(3 + 4i) * (3 - 4i) = 25

(2 + 3i) / (3 + 4i) = (-0.28 + 0.96i)

1 / (3 + 4i) = (0.12 - 0.16i)

false

true

true

3.605551275463989

5

5

小结

如此，复数的四则运算基本都实现了，当然复数还有三角表示式和指数表示式，根据它们的数学定义写出相当代码应该不是很难。有了复数三角式，就能方便地定义出复数的开方运算，有空可以写写这方面的代码。

本文完

文章标签：
Rust
C++
Python
BI
关键词：
Java字符串
Java改写
Rust运算符
Java前缀
HannYang
目录
相关文章
喵手
|
1月前
|
存储 安全 Java
Java零基础-字符串详解
【10月更文挑战第18天】Java零基础教学篇，手把手实践教学！
喵手
103 60
众所周知
|
20天前
|
缓存 算法 Java
本文聚焦于Java内存管理与调优，介绍Java内存模型、内存泄漏检测与预防、高效字符串拼接、数据结构优化及垃圾回收机制
在现代软件开发中，性能优化至关重要。本文聚焦于Java内存管理与调优，介绍Java内存模型、内存泄漏检测与预防、高效字符串拼接、数据结构优化及垃圾回收机制。通过调整垃圾回收器参数、优化堆大小与布局、使用对象池和缓存技术，开发者可显著提升应用性能和稳定性。
众所周知
40 6
java冯坚持
|
2月前
|
Java 数据库
案例一：去掉数据库某列中的所有英文，利用java正则表达式去做，核心：去掉字符串中的英文
这篇文章介绍了如何使用Java正则表达式从数据库某列中去除所有英文字符。
java冯坚持
53 15
小强在此
|
2月前
|
Java
JAVA易错点详解(数据类型转换、字符串与运算符)
JAVA易错点详解(数据类型转换、字符串与运算符)
小强在此
51 4
刘大猫.
|
3月前
|
Java 数据库
java小工具util系列1：日期和字符串转换工具
java小工具util系列1：日期和字符串转换工具
刘大猫.
52 3
刘大猫.
|
3月前
|
SQL Java 索引
java小工具util系列2：字符串工具
java小工具util系列2：字符串工具
刘大猫.
18 2
bzf4tpxf3rljg
|
3月前
|
存储 移动开发 Java
java核心之字符串与编码
java核心之字符串与编码
bzf4tpxf3rljg
25 2
一只牛博
|
7月前
|
存储 XML 缓存
Java字符串内幕：String、StringBuffer和StringBuilder的奥秘
Java字符串内幕：String、StringBuffer和StringBuilder的奥秘
一只牛博
72 0
土木林森
|
4月前
|
安全 Java API
【Java字符串操作秘籍】StringBuffer与StringBuilder的终极对决！
【8月更文挑战第25天】在Java中处理字符串时，经常需要修改字符串，但由于`String`对象的不可变性，频繁修改会导致内存浪费和性能下降。为此，Java提供了`StringBuffer`和`StringBuilder`两个类来操作可变字符串序列。`StringBuffer`是线程安全的，适用于多线程环境，但性能略低；`StringBuilder`非线程安全，但在单线程环境中性能更优。两者基本用法相似，通过`append`等方法构建和修改字符串。
土木林森
67 1
土木林森
|
4月前
|
API C# 开发者
WPF图形绘制大师指南：GDI+与Direct2D完美融合，带你玩转高性能图形处理秘籍！
【8月更文挑战第31天】GDI+与Direct2D的结合为WPF图形绘制提供了强大的工具集。通过合理地使用这两种技术，开发者可以创造出性能优异且视觉效果丰富的WPF应用程序。在实际应用中，开发者应根据项目需求和技术背景，权衡利弊，选择最合适的技术方案。
土木林森
160 0

热门文章

最新文章

  • 1
    Rust 笔记：Rust 语言中使用 vector（向量）
  • 2
    性能的极致,Rust的加持,Zed-Dev编辑器快速搭建Python3.10开发环境
  • 3
    Rust 大展拳脚的新兴领域：机密计算
  • 4
    Rust 笔记：Rust 语言中哈希结构（哈希映射，HashMap）、集合（哈希集，HashSet）及其使用
  • 5
    2021 年 Rust 生态版图调研报告 | 星辰大海（下篇）
  • 6
    Rust 笔记：Rust 语言中的常量与变量
  • 7
    rust高级 异步编程 二 pin（2）
  • 8
    Rust 开发命令行工具（上）（一）
  • 9
    【深入探索Rust：结构体、枚举与模式匹配】A Deep Dive into Rust: Structs, Enums, and Pattern Matching
  • 10
    Rust中的变量与常量：定义与使用的深入解析
  • 1
    面试官：素有Java锁王称号的‘StampedLock’你知道吗？我：这什么鬼？
    99
  • 2
    Java的取余如何编写代码
    75
  • 3
    Java获取字符串最后一位
    564
  • 4
    Java一分钟之-Lambda表达式与函数式接口
    74
  • 5
    Java一分钟之-Java注解的理解与应用
    46
  • 6
    Java一分钟之线程池：ExecutorService与Future
    138
  • 7
    深入理解Java并发编程：线程安全与性能优化
    46
  • 8
    Java 8新特性之Lambda表达式与Stream API的深度解析
    92
  • 9
    JDK、JRE、Java SE、Java EE和Java ME有什么区别?
    131
  • 10
    Java一分钟：线程协作：wait(), notify(), notifyAll()
    64

    • 相关课程

    更多
  • Java面试疑难点解析 - 面试技巧及语言基础
  • Java面试疑难点解析 - Java Web开发
  • Java面试疑难点解析 - 系统架构及项目设计
  • Java编程入门
  • Java面向对象编程
  • Java高级编程

    • 相关电子书

    更多
  • Spring Cloud Alibaba - 重新定义 Java Cloud-Native
  • The Reactive Cloud Native Arch
  • JAVA开发手册1.5.0

    • 相关实验场景

    更多
  • 函数类型-3：函数重载
  • 语言入门-2：基本类型
  • 阿里云平台上进行Java程序的编译与运行
  • 使用Java面向对象编写网络通信程序应用
  • 手动部署Java Web环境（Alibaba Cloud Linux 2）
  • 搭建Java Web开发环境（Anolis OS）
    • 下一篇
    阿里云无影云电脑免费试用，最长可试用3个月