Programmers

다항식 더하기

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

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

문제

코드

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'

728x90