537. 复数乘法(Python)

2020-11-03  本文已影响0人  玖月晴

题目

难度:★★☆☆☆
类型:字符串
方法:无

力扣链接请移步本题传送门
更多力扣中等题的解决方案请移步力扣中等题目录

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

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

解答

这道题不难,先把实数和虚数提取出来,然后按照计算规则计算得到结果,再用相应格式表示即可。

class Solution:
    def complexNumberMultiply(self, a, b):
        a_real, a_imaginary = [int(s) for s in a.strip('i').split('+')]
        b_real, b_imaginary = [int(s) for s in b.strip('i').split('+')]

        ans_real = a_real * b_real - a_imaginary * b_imaginary
        ans_imaginary = a_real * b_imaginary + a_imaginary * b_real
        return '{}+{}i'.format(ans_real, ans_imaginary)

如有疑问或建议,欢迎评论区留言~

有关更多力扣中等题的python解决方案,请移步力扣中等题解析

上一篇 下一篇

猜你喜欢

热点阅读