[카카오 인턴] 수식 최대화

코딩테스트 연습 - 수식 최대화 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

 

나의 코드

from itertools import permutations

def solution(expression):

    answer = 0

    # 부호의 우선순위 경우의 수

    expressioin_list = set()

    for i in expression:

        if i in ["-","+","*"]:

            expressioin_list.add(i)

   

    for i in permutations(expressioin_list,len(expressioin_list)):

        # 부호와 분리해 새로운 리스트를 만들어 주자

        new_expression = []

        str = ""

        for x in expression:

            if x in ['-','+','*']:

                new_expression.append(str)

                new_expression.append(x)

                str = ""

            else:

                str+=x

        new_expression.append(str)

        for j in i:

            # 현재 우선순위가 '-' 인 경우

            if j =='-':

                while True:

                    if '-' not in new_expression:

                        break

                    for i in range(len(new_expression)):

                        if new_expression[i] == '-':

                            new_expression[i] = int(new_expression[i-1]) - int(new_expression[i+1])

                            new_expression.pop(i-1)

                            new_expression.pop(i)

                            break

           

            # 현재 우선순위가 '+' 인 경우

            if j =='+':

                while True:

                    if '+' not in new_expression:

                        break

                    for i in range(len(new_expression)):

                        if new_expression[i] == '+':

                            new_expression[i] = int(new_expression[i-1]) + int(new_expression[i+1])

                            new_expression.pop(i-1)

                            new_expression.pop(i)

                            break

           

            # 현재 우선순위가 '*' 인 경우

            if j =='*':

                while True:

                    if '*' not in new_expression:

                        break

                    for i in range(len(new_expression)):

                        if new_expression[i] == '*':

                            new_expression[i] = int(new_expression[i-1]) * int(new_expression[i+1])

                            new_expression.pop(i-1)

                            new_expression.pop(i)

                            break

            #print(new_expression)

        answer = max(answer, abs(int(new_expression[0])))

    #print(answer)

    return answer

solution("50*6-3*2")