BAEKJOON PYTHON 14888 연산자 끼워넣기

Prob

 

---

Approach

itertools를 사용한다.

음수 // 양수에 주의하여 작성한다.

---

Code

from itertools import permutations
import sys
def cal_formul(case, num):
    res = num[0]
    for i in range(len(num) - 1):
        if case[i] == '+':
            res += num[i + 1]
        elif case[i] == '-':
            res -= num[i + 1]
        elif case[i] == '*':
            res *= num[i + 1]
        elif case[i] == '/':
            if res < 0:
                res = -(abs(res) // num[i + 1])
            else:
                res //= num[i + 1]
    return res

if __name__ == "__main__":
    N = int(input())
    num = list(map(int, input().split()))
    oper = list(map(int, input().split()))
    m_oper = ""

    for i in range(len(oper)):
        if oper[i] > 0 and i == 0:
            m_oper += "+" * oper[i]
        elif oper[i] > 0 and i == 1:
            m_oper += "-" * oper[i]
        elif oper[i] > 0 and i == 2:
            m_oper += "*" * oper[i]
        elif oper[i] > 0 and i == 3:
            m_oper += "/" * oper[i]
    m_oper = list(set((permutations(m_oper, len(m_oper)))))
    min = sys.maxsize
    max = -sys.maxsize - 1
    
    for case in m_oper:
        val = cal_formul(case, num)
        if val < min : min = val
        if val > max : max = val

    print(max)
    print(min)

 

출처

https://www.acmicpc.net/problem/14888