来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/complex-number-multiplication
题目描述
复数 可以用字符串表示,遵循 "实部+虚部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 都是有效的复数表示。
解题思路
一道简单的水题,一觉醒来乌克兰要没了。
根据+ 和i 做分界符将字符串转换成整形,然后根据复数相乘的规则算出结果拼装字符串。
代码展示
class Solution { public: string complexNumberMultiply(string num1, string num2) { string strRet; int iNum[4]; for(int i = 0; i < 2; i++) { int iTemp = 0; string strTemp; for(auto c: i ? num2: num1) { if(c == '+' || c == 'i') { iNum[iTemp + i * 2] = atoi(strTemp.c_str()); iTemp++; strTemp.clear(); } else { strTemp.push_back(c); } } } strRet = to_string(iNum[0] * iNum[2] - iNum[1] * iNum[3]) +"+" + to_string(iNum[0] * iNum[3] + iNum[1] * iNum[2]) + "i"; return strRet; } };
运行结果