[백준14888] 연산자 끼워넣기

문제

---

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

14888번: 연산자 끼워넣기

 

생각의 흐름

---

1. +,-,*,//에 각각 1,2,3,4를 넘버링하여 각 연산자의 갯수만큼 리스트에 채운다.
2. 이들의 순열을 이용하여 가능한 모든 연산의 경우의 수를 구한다.
3. 비교하며 최솟값, 최댓값을 찾는다.

 

코드

---

from itertools import permutations

N = int(input())
A = list(map(int, input().split()))
plus, minus, multiple, division = map(int, input().split())

operation_list = []
operation_list += [1]*plus
operation_list += [2]*minus
operation_list += [3]*multiple
operation_list += [4]*division

operation_set = []
for numbers in list(permutations(operation_list, )):
    operation_set.append(numbers)
operation_set = list(set(operation_set))

max_answer = -1000000001
min_answer = 1000000001
for case in operation_set:
    answer = A[0]

    for i in range(N-1):
        if case[i] == 1:
            answer += A[i+1]
        elif case[i] == 2:
            answer -= A[i+1]
        elif case[i] == 3:
            answer *= A[i+1]
        else:
            if answer<0:
                answer = -(-answer//A[i+1])
            else:
                answer //= A[i+1]
    
    if answer<min_answer:
        min_answer = answer
    if answer>max_answer:
        max_answer = answer

print(max_answer)
print(min_answer)

출처 : https://claude-u.tistory.com/371