[Code Kata] 복소수 곱셈 구현

Q.

두 개의 input에는 복소수(complex number)가 string 으로 주어집니다. 복소수란 a+bi 의 형태로, 실수와 허수로 이루어진 수입니다.

input으로 받은 두 수를 곱해서 반환해주세요. 반환하는 표현도 복소수 형태의 string 이어야 합니다.

A.

def complex_number_multiply(a, b):
  if "+" in a:
    a = a.split("+")

  if "-" in a:
    a = a.split("-")

  if "+" in b:
    b = b.split("+")

  if "-" in b:
    b = b.split("-")

  a[1]=a[1].replace("i","")
  b[1]=b[1].replace("i","")

  x = int(a[0])
  y = int(a[1])
  z = int(b[0])
  w = int(b[1])

  return (str(x*z-y*w)+"+"+str(y*z + x*w)+"i")