동캄의 코딩도장

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

코테/프로그래머스 level0

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

동 캄 2023. 3. 5. 13:12

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

 

프로그래머스

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

programmers.co.kr

#프로그래머스 다항식 더하기
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

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