코테/프로그래머스
프로그래머스 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에 가장 큰 절댓값을 대입한다.
반응형