Link

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

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

programmers.co.kr

Solution

from itertools import permutations

def calc(oper, n, expression):
    if expression.isdigit():
        return str(expression)
    else:
        if oper[n] == '*':
            split_expression = expression.split('*')
            temp = []

            for i in range(len(split_expression)):
                temp.append(calc(oper, n + 1, split_expression[i]))

            return str(eval('*'.join(temp)))

        elif oper[n] == '+':
            split_expression = expression.split('+')
            temp = []

            for i in range(len(split_expression)):
                temp.append(calc(oper, n + 1, split_expression[i]))

            return str(eval('+'.join(temp)))

        elif oper[n] == '-':
            split_expression = expression.split('-')
            temp = []

            for i in range(len(split_expression)):
                temp.append(calc(oper, n + 1, split_expression[i]))

            return str(eval('-'.join(temp)))


def solution(expression):
    answer = 0

    oper = ['*', '+', '-']
    oper = list(permutations(oper, 3))

    for i in range(len(oper)):
        result = abs(int(calc(oper[i], 0, expression)))

        answer = max(answer, result)

    return answer

🔑 key point: Recursive Function
📌 Tip: itertools.permutataions()

Other Solution

from itertools import permutations

def solution(expression):
    oper = ['+', '-', '*']
    oper = list(permutations(oper, 3))
    answer = []

    for op in oper:
        a = op[0]
        b = op[1]
        temp_list = []

        for e in expression.split(a):
            temp = [f'({i})' for i in e.split(b)]
            temp_list.append(f'({b.join(temp)})')

        answer.append(abs(eval(a.join(temp_list))))

    return max(answer)

🔑 key point: [f'({i})' for i in e.split(b)]
📌 Tip: itertools.permutataions()

저작자표시 (새창열림)