LeetCode-537复数乘法

给定两个表示的字符串。
返回表示它们乘积的字符串。注意，根据定义 i² = -1 。
示例 1:

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

示例 2:

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

public class LeetCode537 {

    public static void main(String[] args) {
        String a = "1+-1i";
        String b = "1+-1i";
        String s = complexNumberMultiply(a, b);
        System.out.println(s);
    }
    public static String complexNumberMultiply(String a, String b) {
        StringBuffer buffer = new StringBuffer();
        String[] aArray = a.split("\\+");
        String[] bArray = b.split("\\+");
        int num = Integer.parseInt(aArray[0])*Integer.parseInt(bArray[0])-Integer.parseInt(aArray[1].replace("i", ""))*Integer.parseInt(bArray[1].replace("i", ""));
        int num1 = Integer.parseInt(aArray[0])*Integer.parseInt(bArray[1].replace("i", ""))+Integer.parseInt(aArray[1].replace("i", ""))*Integer.parseInt(bArray[0]);
        return buffer.append(num).append("+").append(num1).append("i").toString();
    }
}