[프로그래머스] Lv.0 다항식 더하기 / JS

📕 문제 설명

한 개 이상의 항의 합으로 이루어진 식을 다항식이라고 합니다. 다항식을 계산할 때는 동류항끼리 계산해 정리합니다. 덧셈으로 이루어진 다항식 polynomial이 매개변수로 주어질 때, 동류항끼리 더한 결괏값을 문자열로 return 하도록 solution 함수를 완성해보세요. 같은 식이라면 가장 짧은 수식을 return 합니다.

⭐ 제한사항

• 0 < polynomial에 있는 수 < 100
• polynomial에 변수는 'x'만 존재합니다.
• polynomial은 양의 정수, 공백, ‘x’, ‘+'로 이루어져 있습니다.
• 항과 연산기호 사이에는 항상 공백이 존재합니다.
• 공백은 연속되지 않으며 시작이나 끝에는 공백이 없습니다.
• 하나의 항에서 변수가 숫자 앞에 오는 경우는 없습니다.
• " + 3xx + + x7 + "와 같은 잘못된 입력은 주어지지 않습니다.
• 0으로 시작하는 수는 없습니다.
• 문자와 숫자 사이의 곱하기는 생략합니다.
• polynomial에는 일차 항과 상수항만 존재합니다.
• 계수 1은 생략합니다.
• 결괏값에 상수항은 마지막에 둡니다.
• 0 < polynomial의 길이 < 50

 

💻 입출력 예

polynomial	result
"3x + 7 + x"	"4x + 7"
"x + x + x"	"3x"

 

📚 입출력 예 설명

입출력 예 #1

• "3x + 7 + x"에서 동류항끼리 더하면 "4x + 7"입니다.

 

입출력 예 #2

• "x + x + x"에서 동류항끼리 더하면 "3x"입니다.

 

😊 나의 답변

function solution(polynomial) {
    // 항만 담긴 배열 생성
    const arr = polynomial.split(" + ");
    const x = [], n = [];

    // 동류항끼리 나누기
    arr.forEach((v) => v.includes("x") ? x.push(v) : n.push(v));

    // x 숫자 연산
    const nOp = x.filter((v) => v.match(/\dx/g)).map((v) => v.substr(0, v.length - 1));
    const sum1 = nOp.reduce((a, c) => a + Number(c), 0);

    // x 연산
    const sum2 = x.length - nOp.length;

    // 상수항 연산
    const sum3 = n.reduce((a, c) => a + Number(c), 0);

    // 출력
    if (polynomial === "x") {
        return "x";
    } else if(sum3 === 0) {
        return (sum1 + sum2) + "x"
    } else {
        if(x.length === 0) return sum3 + "";
        else if((sum1 + sum2) === 1) return "x" + " + " + sum3;
        else return (sum1 + sum2) + "x" + " + " + sum3;
    }
}

연산까지는 제대로 했는데, 출력 시 오류있는 테스트 케이스가 많았다.

 

1. "x"만 있는 경우
2. 일차항만 존재하는 경우
3. 상수항만 존재하는 경우
4. 일차항의 계수가 1인 경우

 

이 부분을 고려하여 풀이해야 했다.

 

⭐ 참고할 답변

function solution(polynomial) {
    const arr = polynomial.split(" + ");
    const xNum = arr
                .filter(n => n.includes("x"))
                .map(n => n.replace('x', '') || '1')
                .reduce((acc, cur) => acc + parseInt(cur, 10), 0);
    const num = arr
                .filter(n => !isNaN(n))
                .reduce((acc, cur) => acc + parseInt(cur, 10), 0);

    let answer = [];
    if(xNum) answer.push(`${xNum === 1 ? "" : xNum}x`);
    if(num) answer.push(num);

    return answer.join(" + ");
}

x의 계수를 따로 배열로 만들어 reduce를 사용하여 값을 구한 것은 나의 풀이와 비슷한데, answer 배열을 생성하여 값을 삽입하고, 출력하니 훨씬 깔끔하고 이해가 쉽다.

 

나는 메서드가 중첩되니까 길어져서 변수에 따로 담았는데 줄바꿈을 이렇게 하니까 보기도 쉽고 이해하기도 쉽다.

다음엔 더 좋은 답안을 작성할 수 있게 노력해야쥐 ✨