[프로그래머스] 다항식 더하기 - 파이썬

Programmers

다항식 더하기

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

프로그래머스

 

문제

코드

• isdigit()을 사용해서 x항이 아닌 경우를 분리
• 계수의 경우에 따라서 return 값 출력

def solution(polynomial):
    xnum = 0
    snum = 0
    for i in polynomial.split(' + '):
        if i.isdigit():
            snum += int(i)
        else:
            if i == 'x':
                xnum += 1
            else:
                xnum += int(i[:-1])

    if xnum == 0:
        return str(snum)
    elif xnum == 1:
        if snum != 0:
            return 'x + ' + str(snum)
        else:
            return 'x'
    else:
        if snum != 0:
            return f'{xnum}x + {snum}'
        else:
            return f'{xnum}x'