[Code Kata] 3.1 complexNumberMultiply

문제

두 개의 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 complexNumberMultiply = (a, b) => {
  // 여기에 코드를 작성해주세요.
  let w = [];
  let x = [];
  let y = [];
  let z = [];
  let i = 0;
  let ret = "";

  while (a[i] !== '+') {
    w.push(a[i]);
    i++;
  }
  while (a[i] !== 'i') {
    x.push(a[i]);
    if (a[i + 1]=== '-') x.pop();
    i++;
  }
  
  i = 0;
  while (b[i] !== '+') {
    y.push(b[i]);
    i++;
  }
  while (b[i] !== 'i') {
    z.push(b[i]);
    if (b[i + 1] === '-') z.pop();
    i++;
  }
  w = Number(w.join(""));
  x = Number(x.join(""));
  y = Number(y.join(""));
  z = Number(z.join(""));

  ret = `${w*y - x*z}+${w*z + x*y}i`
  return (ret);
}