537. Complex Number Multiplication-复数乘法

复数 可以用字符串表示，遵循"实部+虚部i"的形式，并满足下述条件：

• 实部是一个整数，取值范围是[-100, 100]
• 虚部也是一个整数，取值范围是[-100, 100]
• i^2 == -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都是有效的复数表示。

Python:

class Solution:
    def complexNumberMultiply(self, num1: str, num2: str) -> str:
        rea1, ima1 = num1.split('+')
        rea2, ima2 = num2.split('+')
        rea1 = int(rea1)
        rea2 = int(rea2)
        ima1 = int(ima1[:len(ima1)-1])
        ima2 = int(ima2[:len(ima2)-1])
        ret1 = rea1*rea2 - ima1*ima2
        ret2 = rea1*ima2 + rea2*ima1
        return str(ret1)+'+'+str(ret2)+'i'

Java:

class Solution {
    public String complexNumberMultiply(String num1, String num2) {
        String[] nums1 = num1.split("\\+");
        String[] nums2 = num2.split("\\+");
        int rea1 = Integer.parseInt(nums1[0]);
        int ima1 = Integer.parseInt(nums1[1].substring(0, nums1[1].length()-1));
        int rea2 = Integer.parseInt(nums2[0]);
        int ima2 = Integer.parseInt(nums2[1].substring(0, nums2[1].length()-1));
        int ret1 = rea1*rea2 - ima1*ima2;
        int ret2 = rea1*ima2 + rea2*ima1;
        return ret1+"+"+ret2+"i";
    }
}