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

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

프로그래머스

#프로그래머스 다항식 더하기
def solution(polynomial):
    x_cnt=0
    digit=0
    arr=list(map(str,polynomial.split()))
    for val in arr:
        if val!='+':
            if 'x' in val:
                if len(val)==1:
                    x_cnt+=1
                else:
                    x_cnt+=int(val[:-1])
            else:
                digit+=int(val)
    if x_cnt==0:
        answer="{}".format(digit)
    elif x_cnt==1:
            if digit!=0:
                answer="x + {}".format(digit)
            else:
                answer="x"
    elif x_cnt>=1:
        if digit!=0:
            answer="{}x + {}".format(x_cnt,digit)
        else:
            answer="{}x".format(x_cnt)
    return answer

여러 케이스를 개별로 생각해서 해결한다.