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")