[BOJ/Step14] 14888 : 연산자 끼워넣기 (Python)

www.acmicpc.net/problem/14888

14888번: 연산자 끼워넣기

---

입력받은 연산자 개수를 어떻게 배치하느냐가 관건..

일단 순열을 모두 구한 다음에 집합으로 변경해 중복을 날려버리기로 했다

처음에는 숫자 + 연산자 + 숫자 해서 쭉 str로 만들어 eval 함수로 계산했는데

음수 // 양수 하면 값이 달라지는것... 그래서 어쩔수 없이 하나하나 계산했다

 

연산자와 숫자 리스트를 전달하면 계산해서 결과값을 반환하는 함수 expr

  1. num의 인덱스 idx = 0

  2. 계산한 결과값을 저장할 total 은 첫번째 숫자로 초기화

  3. 연산자 리스트 ope에서 하나씩 가져온다

    3-1. 연산자가 +, -, * 일 때는 그대로 계산

    3-2. 연산자가 / 일 때는 첫번째 피연산자 = total 이 음수인지 확인해서 따로 계산해준다 (양수로 바꾼 뒤 몫을 취하고, 그 몫을 음수로 바꾼 것)

  4. 결과값 반환

 

1. 숫자와 연산자 개수를 각각 nums, op1 리스트에 저장한다

2. 연산자 개수에 각각 +, -, *, / 를 곱해 하나의 문자열 op2로 만든다

3. itertools 모듈의 permutation() 함수를 사용해 op2를 사용해 순열 조합을 생성해 집합 operator으로 저장한다 ( 중복 제거 )

4. 집합 operator을 리스트로 변경한다

5. operator의 요소를 각각 계산하여 결과값을 저장하는 리스트 result에 추가

6. result의 최댓값과 최솟값 출력

 

# 14888.py

import itertools


def expr(ope, num):
    idx = 0
    total = num[idx]
    for i in ope:
        if i == '+':
            total = total + num[idx+1]
        elif i == '-':
            total = total - num[idx+1]
        elif i == '*':
            total = total * num[idx+1]
        elif i == '/':
            if total < 0:
                total = (abs(total) // num[idx+1]) * -1
            else:
                total = total // num[idx+1]
        idx += 1
    return total


n = int(input())
nums = list(map(int, input().split()))
op1 = list(map(int, input().split()))

op2 = '+' * op1[0] + '-' * op1[1] + '*' * op1[2] + '/' * op1[3]
operator = set(itertools.permutations(op2, n - 1))
operator = list(operator)
operator.sort()

result = []
for i in operator:
    result.append(expr(i, nums))

print(max(result))
print(min(result))