LeetCode 537. Complex Number Multiplication

Description

Given two strings representing two complex numbers.

You need to return a string representing their multiplication. Note $i^2 = -1$ according to the definition.
Example 1:

Input: “1+1i”, “1+1i”
Output: “0+2i”
Explanation: (1 + i) * (1 + i) = 1 + i2 + 2 * i = 2i, and you need convert it to the form of 0+2i.

Example 2:

Input: “1+-1i”, “1+-1i”
Output: “0+-2i”
Explanation: (1 - i) * (1 - i) = 1 + i2 - 2 * i = -2i, and you need convert it to the form of 0+-2i.

**Note:
The input s**trings will not have extra blank.
The input strings will be given in the form of a+bi, where the integer a and b will both belong to the range of [-100, 100]. And the output should be also in this form.
Subscribe to see which companies asked this question.

Code

class Solution {
public:
    string complexNumberMultiply(string a, string b) {
        int xa=stoi(a.substr(0,a.find('+')));
        int xb=stoi(b.substr(0,b.find('+')));
        int ya=stoi(a.substr(a.find('+')+1,a.find('i')-a.find('+')-1));
        int yb=stoi(b.substr(b.find('+')+1,b.find('i')-b.find('+')-1));
        string result = to_string(xa*xb-ya*yb) + "+" + to_string(xa*yb+xb*ya) + "i";
        return result;
    }
};

Appendix

• Link: https://leetcode.com/problems/complex-number-multiplication/
• Run Time: 3ms