[BOJ/백준] 14888. 연산자 끼워넣기 (Python)

7691 단어 DFS브루트포스DFS

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

Problem

수열이 차례 대로 주어지고 연산자 +-*/의 각각의 개수가 차례대로 주어질때 최댓값, 최솟값을 찾는 문제

Solution

DFS를 통해 모든 경우의 수를 탐색.

단 여기서, 나누기의 경우에 좀 예외적인 경우라 파이썬 기준으로 int(a/b)로 해야함 a//b로 하면 틀림. 예제 3의 힌트를 보면 이를 알수 있다.

Python Code

import sys

N=int(sys.stdin.readline())

arrList=list(map(int,sys.stdin.readline().split()))
advisor=list(map(int,sys.stdin.readline().split()))
maxValue=-1e9
minValue=1e9
def dfs(initationValue,depth,plus,minus,multiply,divide):
    global maxValue
    global minValue
    if depth==N:
        maxValue=max(maxValue,initationValue)
        minValue=min(minValue,initationValue)

    if plus:
        dfs(initationValue+arrList[depth],depth+1,plus-1,minus,multiply,divide)
    if minus:
        dfs(initationValue-arrList[depth],depth+1,plus, minus-1, multiply, divide)
    if multiply:
        dfs(initationValue*arrList[depth],depth+1,plus, minus, multiply-1, divide)
    if divide:
        dfs(int(initationValue/arrList[depth]),depth+1,plus, minus, multiply, divide-1)

dfs(arrList[0],1,advisor[0],advisor[1],advisor[2],advisor[3])

print(maxValue)
print(minValue)


좋은 웹페이지 즐겨찾기