LeetCode - 537. 复数乘法

描述

给定两个表示复数的字符串。

返回表示它们乘积的字符串。注意，根据定义 i2 = -1 。

示例 1:

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

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

输入字符串不包含额外的空格。
输入字符串将以 a+bi 的形式给出，其中整数 a 和 b 的范围均在 [-100, 100] 之间。输出也应当符合这种形式。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/complex-number-multiplication/
 

求解

class Solution {
public:
    // 方法一，根据公式暴力计算
    string complexNumberMultiply_1e(string a, string b) {
        auto nums1 = getNumber(a);
        auto nums2 = getNumber(b);
        int &x = nums1[0];
        int &y = nums1[1];
        int &m = nums2[0];
        int &n = nums2[1];
        int pre = x * m - y * n;
        int post = x * n + y * m;
        string res = to_string(pre) + "+" + to_string(post) + "i";
        return res;
    }

    // 方法二，直接使用标准库complex类
    string complexNumberMultiply(string a, string b) {
        auto c1 = parseComplex(a);
        auto c2 = parseComplex(b);
        auto res = c1 * c2;
        return to_string(res.real()) + "+" + to_string(res.imag()) + "i";
    }

private:
    // 解析实部和虚部数值
    vector<int> getNumber(const string &str) {
        auto plusPos = str.find('+');
        string num1 = str.substr(0, plusPos);
        string num2 = str.substr(plusPos + 1, str.size() - plusPos - 2);
        vector<int> nums;
        nums.emplace_back(stoi(num1));
        nums.emplace_back(stoi(num2));
        return nums;
    }

    complex<int> parseComplex(const string &str) {
        auto plusPos = str.find('+');
        int real = stoi(str.substr(0, plusPos));
        int imag = stoi(str.substr(plusPos + 1, str.size() - plusPos - 2));
        return {real, imag};
    }
};