[프로그래머스 | 파이썬 / 자바스크립트] 다항식 더하기(코딩테스트 입문/ level 0)

문제

https://school.programmers.co.kr/learn/courses/30/lessons/120863?language=python3 

프로그래머스

 

 

🐍파이썬

def solution(polynomial):
    arr = polynomial.replace(" ", "").split("+")
    x, num = 0, 0
    for i in arr:
        if "x" in i:
            if len(i) >= 2:
                x += int(i[:-1])
            else:
                x += 1
        else:
            num += int(i)
    if num > 0 and x > 0:
        if x == 1:
            return "x + "+str(num)
        else:
            return str(x)+"x + "+str(num)
    elif x <= 0:
        return str(num)
    elif num <= 0:
        if x == 1:
            return "x"
        else:
            return str(x)+"x"

1️⃣ 문자열을 "+" 기준으로 분리

2️⃣ 분리한 문자를 배열에 담아 for문에서 하나씩 확인

3️⃣ 만약 배열의 원소가 "x"를 포함하고 있다면 'x항'이라는 뜻. 'x항'의 길이가 2보다 길다면 x의 앞부분에 정수가 온다는 의미. 정수는 분리해서 int로 형변환하여 변수 x에 더해준다.

4️⃣ x를 포함하지 않은 상수항도 num 변수에 따로 저장

⛔ for문 종료

5️⃣ 상황별로 적절한 처리를 통해 결과를 도출한다. 13번 라인, 20번 라인처럼 x항의 계수가 1일 경우를 반드시 따로 처리해 주어야 함을 주의

 

 

다른 풀이 방법

def solution(polynomial):
    xnum = 0
    const = 0
    for c in polynomial.split(' + '):
        if c.isdigit():
            const+=int(c)
        else:
            xnum = xnum+1 if c=='x' else xnum+int(c[:-1])
    if xnum == 0:
        return str(const)
    elif xnum==1:
        return 'x + '+str(const) if const!=0 else 'x'
    else:
        return f'{xnum}x + {const}' if const!=0 else f'{xnum}x'

 

 

🐥자바스크립트

function solution(polynomial) {
    let arr = polynomial.split(" + ")
    let x = 0
    let num = 0
    for(let i of arr){
        if(i.includes("x")){
            i === "x" ? x += 1 : x += parseInt(i);
        } else {
            num += parseInt(i);
        }
    }
    if(x > 0 && num > 0){
        return x === 1 ? "x + " + num.toString() : x.toString() + "x + " + num.toString();
    } else if(x == 0){
        return num.toString();
    } else if(num == 0){
        return x === 1 ? "x" : x.toString() + "x";
    }
}

1️⃣ 문자열을 " + " 기준으로 분리

2️⃣ 분리한 문자를 배열에 담아 for문에서 하나씩 확인

3️⃣ 만약 배열의 원소가 "x"를 포함하고 있다면 'x항'이라는 뜻. i가 x라면 계수가 1이라는 의미이므로 x에 1을 더해준다. x의 계수가 1이 아닌 경우 정수는 분리해서 parseInt로 형변환하여 변수 x에 더해준다.

4️⃣ x를 포함하지 않은 상수항도 num 변수에 따로 저장

⛔ for문 종료

5️⃣ 상황별로 적절한 처리를 통해 결과를 도출한다. x항의 계수가 1일 경우를 반드시 따로 처리해 주어야 함을 주의

 

 

다른 풀이 방법

function solution(polynomial) {
    const arr = polynomial.split(" + ");
    const xNum = arr
                .filter(n => n.includes("x"))
                .map(n => n.replace('x', '') || '1')	// '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(" + ");
}

• filter(),  map(),  reduce()를 사용한 풀이

🔆 조건문에서 논리연산자(&&, ||) 사용

a AND b
a, b 둘 모두 true일 때 true
a가 true이면 b도 true인지 확인해야 하므로 b문장이 실행
a가 true일때 b를 수행하기 위해 사용

let arr = [43, 64, 77, 50, 1];
arr.map(num => 
   (num % 2 == 0) && console.log(num + ": 2의 배수") 
);​

a OR b
a, b 둘 중 하나만 true이면 true
a가 true이면 b를 확인할 필요없이 true이므로 b를 수행하지 않고 건너뛴다.
a가 false이면 b를 확인할 필요가 있으므로 b를 수행한다.
a가 false일 때 b를 수행하기 위해 사용

let arr = [43, 64, 77, 50, 1];
arr.map(num => 
   (num % 2 == 0) || console.log(num + ": 2의 배수가 아니다") 
);

참고한 블로그: https://tinyurl.com/2lt5dajy