leetcode-537:复数乘法
题目
题目链接
复数 可以用字符串表示,遵循 “实部+虚部i” 的形式,并满足下述条件:
- 实部 是一个整数,取值范围是 [-100, 100]
- 虚部 也是一个整数,取值范围是 [-100, 100]
- i^2 == -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 的形式。
解题
方法一:模拟
struct Num{
int r;
int i;
};
class Solution {
public:
string complexNumberMultiply(string num1, string num2) {
Num n1=getNum(num1);
Num n2=getNum(num2);
return to_string(n1.r*n2.r+n1.i*n2.i*(-1))+'+'+to_string(n1.r*n2.i+n1.i*n2.r)+'i';
}
//获取实部、虚部
Num getNum(string s){
int n=s.size();
int i=0;
while(i<n&&s[i]!='+') i++;//以'+'作为分隔,区别实部和虚部
Num num;
num.r=stoi(s.substr(0,i));
int start=i+1;//跳过'+'
while(i<n&&s[i]!='i') i++;
num.i=stoi(s.substr(start,i-start));
return num;
}
};