leetcode练习题537. 复数乘法

最新推荐文章于 2022-02-25 09:18:21 发布

原创最新推荐文章于 2022-02-25 09:18:21 发布

· 852 阅读

0 ·

版权

给定两个复数字符串，该问题要求计算它们的乘积并以字符串形式返回。通过提取每个字符串的实部和虚部，利用复数乘法规则进行计算。辅助函数geta和getb用于获取复数的系数，然后应用公式求解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

返回表示它们乘积的字符串。注意，根据定义 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] 之间。输出也应当符合这种形式。

思路：

这道题只要把每个字符串里的a与b提取出来，就可以将结果用公式算出来，并转化为字符串返回。

代码：

def geta(string):
    size = len(string)
    pos = 0
    for i in range(0,size):
        if string[i] == '+':
            break
        pos = pos + 1
    return int(string[:pos])

def getb(string):
    size = len(string)
    pos = 0
    for i in range(0,size):
        if string[i] == '+':
            break
        pos = pos + 1
    return int(string[pos+1:-1])    

class Solution:
    
    def complexNumberMultiply(self, a, b):
        """
        :type a: str
        :type b: str
        :rtype: str
        """
        aa = geta(a)
        ab = getb(a)
        ba = geta(b)
        bb = getb(b)
        string = ""
        string = string + str(aa*ba - ab*bb) + "+" + str(aa*bb + ab*ba) + "i"
        return string

代码解析：

这里写了两个辅助函数geta以及getb，分别从复数字符串中获得实数部分以及复数部分，由于题目所给的要求较弱，只要找出'+'的位置，再通过int（）函数将整数字符串转化为整数，即可分别找出复数字符串的两个系数，假设两个复数表达式分别为：

a+bi，c+di，则整数部分为（ac-bd),复数部分为（ad+bc)i，通过公式计算结果返回即可。

提交记录：