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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/complex-number-multiplication
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路及方法

获取两个字符串的实部虚部进行计算就完事了，问题在如何获取各自的实部虚部。我之前本来是根据字符串长度然后获取实部虚部，但是太麻烦了。仔细观察字符串，发现结构为a+b*i，a为实数，b为虚数。所以就直接根据+号将字符串分割为两半，就能很方便取得实部和虚部了。

class Solution {
    public String complexNumberMultiply(String num1, String num2) {
        // num1和nums2的实部虚部
        String[] nums1Str = num1.split("\\+");
        String[] nums2Str = num2.split("\\+");
        int a1 = Integer.parseInt(nums1Str[0]);
        int a2 = Integer.parseInt(nums2Str[0]);
        // 去掉i
        int b1 = Integer.parseInt(nums1Str[1].substring(0, nums1Str[1].length() - 1));
        int b2 = Integer.parseInt(nums2Str[1].substring(0, nums2Str[1].length() - 1));
        // 计算
        int real = a1 * a2 - b1 * b2;
        int imaginary = a1 * b2 + a2 * b1;

        return String.valueOf(real) + "+" + String.valueOf(imaginary) + "i";
    }
}

结果如下：