537. 复数乘法
题目描述
复数 可以用字符串表示,遵循 “实部+虚部i” 的形式,并满足下述条件:
- 实部 是一个整数,取值范围是 [-100, 100]
- 虚部 也是一个整数,取值范围是 [-100, 100]
- i2 == -1
给你两个字符串表示的复数 num1 和 num2 ,请你遵循复数表示形式,返回表示它们乘积的字符串。
示例 1:
输入:num1 = "1+1i", num2 = "1+1i"
输出:"0+2i"
解释:(1 + i) * (1 + i) = 1 + i2 + 2 * i = 2i ,你需要将它转换为 0+2i 的形式。
示例 2:
输入:num1 = "1+-1i", num2 = "1+-1i"
输出:"0+-2i"
解释:(1 - i) * (1 - i) = 1 + i2 - 2 * i = -2i ,你需要将它转换为 0+-2i 的形式。
提示:
- num1 和 num2 都是有效的复数表示。
题解思路
将num1 和 num2 中的实数与虚数部分分离出来
再分别将其相乘 最后返回
题解代码
class Solution:
def complexNumberMultiply(self, num1: str, num2: str) -> str:
true1,xu1=map(int,(num1[:-1].split('+')))
true2,xu2=map(int,(num2[:-1].split('+')))
return f'{true1*true2-xu1*xu2}+{true1*xu2+true2*xu1}i'