537. Complex Number Multiplication**
https://leetcode.com/problems/complex-number-multiplication/
题目描述
Given two strings representing two complex numbers.
You need to return a string representing their multiplication. Note i 2 = − 1 i^2 = -1 i2=−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 strings will not have extra blank.
- The input strings will be given in the form of
a+bi
, where the integera
andb
will both belong to the range of[-100, 100]
. And the output should be also in this form.
C++ 实现 1
复数的乘法: (a + bi) * (c + di) = (ac - bd) + (ad + bc)i
.
class Solution {
private:
vector<int> convert_complex_to_vector(const string &s) {
auto plus_pos = s.find('+');
auto str_a = s.substr(0, plus_pos);
auto str_b = s.substr(plus_pos + 1, std::string::npos - plus_pos - 1);
auto a = std::stoi(str_a), b = std::stoi(str_b);
return {a, b};
}
vector<int> multiply(const vector<int> &v1, const vector<int> &v2) {
auto a = v1[0], b = v1[1], c = v2[0], d = v2[1];
auto r = a * c - b * d;
auto i = a * d + b * c;
return {r, i};
}
string convert_vector_to_complex(const vector<int> &v) {
auto a = v[0], b = v[1];
string s = std::to_string(a) + "+" + std::to_string(b) + "i";
return s;
}
public:
string complexNumberMultiply(string a, string b) {
auto v1 = convert_complex_to_vector(a);
auto v2 = convert_complex_to_vector(b);
auto vec_res = multiply(v1, v2);
auto res = convert_vector_to_complex(vec_res);
return res;
}
};
C++ 实现 2
两年前的代码
class Solution {
private:
pair<int, int> stringToPair(string &comp) {
auto first = comp.find('+');
int a = std::stoi(comp.substr(0, first));
int b = std::stoi(comp.substr(first+1, comp.size() - first - 2));
return make_pair(a, b);
}
public:
string complexNumberMultiply(string a, string b) {
auto p1 = stringToPair(a);
auto p2 = stringToPair(b);
string res = "";
res += to_string(p1.first * p2.first - p1.second * p2.second);
res += "+";
res += to_string(p1.first * p2.second + p1.second * p2.first);
res += "i";
return res;
}
};