LeetCode 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 都是有效的复数表示。

public static String complexNumberMultiply(String num1, String num2) {
    String[] ss1 = num1.split("\\+|i");
    String[] ss2 = num2.split("\\+|i");
    int a = Integer.valueOf(ss1[0]), b = Integer.valueOf(ss1[1]);
    int c = Integer.valueOf(ss2[0]), d = Integer.valueOf(ss2[1]);

    int A = a * c - b * d, B = (a * d) + (b * c);
    return A + "+" + B + "i";
}

测试用例

public static void main(String[] args) {
    String num1 = "1+1i", num2 = "1+1i";
    String anStr = ComplexNumberMultiply.complexNumberMultiply(num1, num2);
    System.out.println("ComplexNumberMultiply.main demo02 result : " + anStr);

    num1 = "1+-1i";
    num2 = "1+-1i";
    anStr = ComplexNumberMultiply.complexNumberMultiply(num1, num2);
    System.out.println("ComplexNumberMultiply.main demo01 result : " + anStr);
}