Question

537. 复数乘法

English Version

题目描述

复数 可以用字符串表示，遵循 "实部+虚部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 都是有效的复数表示。

解法

方法一

class Solution:
  def complexNumberMultiply(self, num1: str, num2: str) -> str:
    a, b = map(int, num1[:-1].split('+'))
    c, d = map(int, num2[:-1].split('+'))
    return f'{a * c - b * d}+{a * d + c * b}i'

class Solution {
  public String complexNumberMultiply(String num1, String num2) {
    String[] c1 = num1.split("\\+|i");
    String[] c2 = num2.split("\\+|i");
    int a = Integer.parseInt(c1[0]);
    int b = Integer.parseInt(c1[1]);
    int c = Integer.parseInt(c2[0]);
    int d = Integer.parseInt(c2[1]);
    return String.format("%d+%di", a * c - b * d, a * d + c * b);
  }
}

class Solution {
public:
  string complexNumberMultiply(string num1, string num2) {
    int a, b, c, d;
    sscanf(num1.c_str(), "%d+%di", &a, &b);
    sscanf(num2.c_str(), "%d+%di", &c, &d);
    return string(to_string(a * c - b * d) + "+" + to_string(a * d + c * b) + "i");
  }
};

func complexNumberMultiply(num1, num2 string) string {
  parse := func(num string) (a, b int) {
    i := strings.IndexByte(num, '+')
    a, _ = strconv.Atoi(num[:i])
    b, _ = strconv.Atoi(num[i+1 : len(num)-1])
    return
  }
  a, b := parse(num1)
  c, d := parse(num2)
  return fmt.Sprintf("%d+%di", a*c-b*d, a*d+b*c)
}

function complexNumberMultiply(num1: string, num2: string): string {
  let arr1 = num1.split('+'),
    arr2 = num2.split('+');
  let r1 = Number(arr1[0]),
    r2 = Number(arr2[0]);
  let v1 = Number(arr1[1].substring(0, arr1[1].length - 1)),
    v2 = Number(arr2[1].substring(0, arr2[1].length - 1));
  let ansR = r1 * r2 - v1 * v2;
  let ansV = r1 * v2 + r2 * v1;
  return `${ansR}+${ansV}i`;
}