LeetCode - Complex Number Multiplication

题目

　　输入两个 a+bi格式的字符串，计算它们的乘积，i^2=-1，输出结果也为a+bi的形式。
　　如 输入”1+1i”, “1+1i” (1 + i) * (1 + i) = 1 + i^2 + 2 * i = 2i，输出0+2i

思路

　　从字符串中抽取出数字进行计算，计算过后再进行拼装结果字符串。

代码

    public String complexNumberMultiply(String a, String b) {
        if (a == null || b == null || a.equals("") || b.equals("")) {
            return "";
        }
        // 将输入用 + 号分割开
        String[] tmpa = a.split("\\+");
        String[] tmpb = b.split("\\+");
        // 提取出所有的数字
        int numA1 = Integer.valueOf(tmpa[0]);
        int numA2 = Integer.valueOf(tmpa[1].replaceAll("i", ""));
        int numB1 = Integer.valueOf(tmpb[0]);
        int numB2 = Integer.valueOf(tmpb[1].replaceAll("i", ""));
        int res1 = -1 * numA2 * numB2 + numA1 * numB1;
        int res2 = numA1 * numB2 + numA2 * numB1;
        return res1 + "+" + res2 + "i";
    }