【Leetcode_总结】537. 复数乘法 - python

Q:

给定两个表示复数的字符串。

返回表示它们乘积的字符串。注意，根据定义 i2 = -1 。

示例 1:

输入: "1+1i", "1+1i"
输出: "0+2i"
解释: (1 + i) * (1 + i) = 1 + i2 + 2 * i = 2i ，你需要将它转换为 0+2i 的形式。

示例 2:

输入: "1+-1i", "1+-1i"
输出: "0+-2i"
解释: (1 - i) * (1 - i) = 1 + i2 - 2 * i = -2i ，你需要将它转换为 0+-2i 的形式。 

---

链接：https://leetcode-cn.com/problems/complex-number-multiplication/description/

思路：暴力求解

代码：

class Solution(object):
    def complexNumberMultiply(self, a, b):
        """
        :type a: str
        :type b: str
        :rtype: str
        """
        a1 = int(a.split("+")[0])
        a2 = int(a.split("+")[1].rsplit("i")[0])
        b1 = int(b.split("+")[0])
        b2 = int(b.split("+")[1].rsplit("i")[0])

        if a1 * b1 == 0:
            res1 = "0"
        else:
            res1 = str(a1 * b1)

        if a1 * b2 + b1 * a2 == 0:
            res2 = "0"
        else:
            res2 = str(a1 * b2 + b1 * a2)

        if a2 * b2 > 0:
            res3 = "-" + str(a2 * b2)
        elif a2 * b2 == 0:
            res3 = "0"
        else:
            res3 = str(- a2 * b2)

        if int(res1) + int(res3) > 0:
            A = str(int(res1) + int(res3))
        elif int(res1) + int(res3) == 0:
            A = "0"
        else:
            A = str(int(res1) + int(res3))
        return A + "+" + res2 + "i"