[python] 프로그래머스 다항식 더하기

/date 24.10.08.

 

 

https://school.programmers.co.kr/learn/courses/30/lessons/120863

프로그래머스

 

def solution(polynomial):
    arr = polynomial.split()
    num, n = 0, 0
    for i in arr:
        if 'x' in i:
            if i == 'x':
                num += 1
            else:
                num += int(i[:-1])
        elif i == '+': 
            continue
        else:
            n += int(i)
    if num == 0 and n != 0:
        return str(n)
    elif num != 0 and n == 0:
        return f"{num}x" if num != 1 else "x"
    else:
        return f"{num}x + {n}" if num != 1 else f"x + {n}"

 

✔️ x 앞의 계수가 1인 경우 1x가 아니라 x를 출력해야 함에 주의

 

✔️x 앞의 계수가 두 자리 이상의 수일 경우 slicing 범위에 주의

 

📌 %{}와 비슷한 기능을 파이썬에서는 f가 해주고 있구나!