동캄의 코딩도장

프로그래머스 level2 [수식 최대화] 파이썬 본문

코테/프로그래머스

프로그래머스 level2 [수식 최대화] 파이썬

동 캄 2021. 12. 23. 21:41

https://programmers.co.kr/learn/courses/30/lessons/67257

 

코딩테스트 연습 - 수식 최대화

IT 벤처 회사를 운영하고 있는 라이언은 매년 사내 해커톤 대회를 개최하여 우승자에게 상금을 지급하고 있습니다. 이번 대회에서는 우승자에게 지급되는 상금을 이전 대회와는 다르게 다음과

programmers.co.kr

#프로그래머스 수식 최대화
def solution(expression):
    from itertools import permutations
    answer = 0
    lst=[]
    j=0
    for i in range(len(expression)):
        if not expression[i].isdigit():
            lst.append(expression[j:i])
            lst.append(expression[i])
            j=i+1
    lst.append((expression[j:]))
    permu=permutations([1,2,3],3)
    for a,b,c in permu:
        p={'+':a,'-':b,'*':c}
        number=[]
        symbol=[]
        for val in lst:
            if val.isdigit():
                number.append(int(val))
            else:
                if symbol:
                    if p[symbol[-1]]>=p[val]:
                        while symbol and p[symbol[-1]]>=p[val]:
                            x=symbol.pop()
                            r=number.pop()
                            l=number.pop()
                            if p[x]==a:
                                number.append(l+r)
                            elif p[x]==b:
                                number.append(l-r)
                            elif p[x]==c:
                                number.append(l*r)
                symbol.append(val)
        while symbol:
            x=symbol.pop()
            r=number.pop()
            l=number.pop()
            if p[x]==a:
                number.append(l+r)
            elif p[x]==b:
                number.append(l-r)
            elif p[x]==c:
                number.append(l*r)
        answer=max(answer,abs(number[0]))

    return answer

먼저, 수식을 분리하고, permutation 함수를 이용해서 연산자 (+,-,*)의 우선순위를 정해준다. 각 연산을 실행하면서, 자기보다 우선순위가 큰 연산자가 오면, 작은 연산자가 나오거나 symbol리스트가 빌 때까지 연산자와 숫자를 pop 하여 연산한다.

answer에 가장 큰 절댓값을 대입한다.