【LeetCode】复数乘法

题目描述

复数 可以用字符串表示，遵循 “实部+虚部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 的形式。

方法

依次拿到复数的实部和虚部，遵循复数乘法运算计算出实部和虚部，组合成字符串然后返回。

代码

class Solution:
    def complexNumberMultiply(self, num1: str, num2: str) -> str:
        a, b, c, d = 0, 0, 0, 0
        mid = ''
        for i in num1:
            if i =='+':
                a = int(mid)
                mid = ''
            if i == 'i':
                b = int(mid[1:])
            mid += i
        mid = ''        
        for i in num2:
            if i =='+':
                c = int(mid)
                mid = ''
            if i == 'i':
                d = int(mid[1:])
            mid += i
        return str(a * c - b * d) + '+' + str(a * d + b * c) + 'i'