Algorithm - CodeKata #11 📌

1. Question

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

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

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

예제 1:
Input: "1+1i", "1+1i"
Output: "0+2i"
설명:
(1 + i) * (1 + i) = 1 + i + i + i^2 = 2i
2i를 복소수 형태로 바꾸면 0+2i.

예제 2:
Input: "1+-1i", "1+-1i"
Output: "0+-2i"
설명:
(1 - i) * (1 - i) = 1 - i - i + i^2 = -2i,
-2i를 복소수 형태로 바꾸면 0+-2i.

예제 3:
Input: "1+3i", "1+-2i"
Output: "7+1i"
설명:
(1 + 3i) * (1 - 2i) = 1 - 2i + 3i -6(i^2) = 1 + i + 6,
7+i를 복소수 형태로 바꾸면 7+1i.

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

2. Answer

const complexNumberMultiply = (a, b) => {
    let a_container = a.split(/[-,+]/);
    let b_container = b.split(/[-,+]/);
    if(a_container.indexOf('') > -1){
      const empty_a = a_container.indexOf('');
      a_container.splice(empty_a,1)
    } 
    if(b_container.indexOf('') > -1){
      const empty_b = b_container.indexOf('');
      b_container.splice(empty_b,1)
    }
    a_container[1] = a_container[1].slice(0,a_container[1].length-1)
    b_container[1] = b_container[1].slice(0,b_container[1].length-1)
    a_container = a_container.map(Number);
    b_container = b_container.map(Number);
    let a_operator = [];
    let b_operator = [];
    for(let i = 0; i < a.length; i++){
        if(a[i] === '+' || a[i] === '-'){
            a_operator.push(a[i])
        }   
    }
    for(let i = 0; i < b.length; i++){
        if(b[i] === '+' || b[i] === '-'){
            b_operator.push(b[i])
        }   
    }

    if(a_operator.length === 2){
        a_container[1] = -1 * a_container[1]
    }
    if(b_operator.length === 2){
        b_container[1] = -1 * b_container[1]
    }

    let container1 = []; // 첫번쨰: 정수끼리, 두번째: 정수*i, 세번째: i의 제곱 1 + i + i + i^2 = 2i 
    container1[0] = a_container[0] * b_container[0]
    container1[1] = a_container[0] * b_container[1] + b_container[0] * a_container[1]
    container1[2] = a_container[1] * b_container[1]
    
    let result = [];
    result[0] = container1[0] - container1[2]
    result[1] = container1[1]
    
    console.log(result) // [-3, 4]
    return `${result[0]}+${result[1]}i`
}

console.log(complexNumberMultiply("1+2i", "1-2i")); // '-3+4i'

출처