CodeKata. 복소수 곱해서 반환하기

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

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

복소수 정의에 의하면 (i^2)는 -1 이므로 (i^2) 일때는 -1로 계산해주세요.

(제곱 표현이 안 되어 i의 2제곱을 (i^2)라고 표현했습니다.)

 

예제 1:
Input: "1+1i", "1+1i"
Output: "0+2i"
Input: "1+1i", "1+1i"
Output: "0+2i"
설명:

(1 + i) * (1 + i) = 1 + i + i + i^2 = 2i 
(1 + i) * (1 + i) = 1 + i + i + i^2 = 2i 
2i를 복소수 형태로 바꾸면 0+2i.

 

예제 2:
Input: "1+-1i", "1+-1i"
Output: "0+-2i"
Input: "1+-1i", "1+-1i"
Output: "0+-2i"
설명:

(1 - i) * (1 - i) = 1 - i - i + i^2 = -2i, 
(1 - i) * (1 - i) = 1 - i - i + i^2 = -2i, 
-2i를 복소수 형태로 바꾸면 0+-2i.

 

예제 3:
Input: "1+3i", "1+-2i"
Output: "7+1i"
Input: "1+3i", "1+-2i"
Output: "7+1i"
설명:

(1 + 3i) * (1 - 2i) = 1 - 2i + 3i -6(i^2) = 1 + i + 6, 
(1 + 3i) * (1 - 2i) = 1 - 2i + 3i -6(i^2) = 1 + i + 6, 
7+i를 복소수 형태로 바꾸면 7+1i.

 

가정
input은 항상 a+bi 형태입니다. output도 a+bi 형태로 나와야 합니다.

const a = "1+2i"
const b = "1+-1i"

const complexNumberMultiply = (a, b) => {
  let firstArr = a.substring(0, a.length-1).split('+');
  const A = Number(firstArr[0])
  console.log(A)
  const B = Number(firstArr[1])
  console.log(B)

  let secondArr = b.substring(0, b.length-1).split('+');
  const C = Number(secondArr[0])
  const D = Number(secondArr[1])

  return `${A * C - B * D}+${A * D + B * C}i`
}

complexNumberMultiply(a, b)