BOJ_16198 [ 에너지 모으기 ] [ 파이썬 ]

문제

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

16198번: 에너지 모으기

문제 해결 방법

처음 그리디로 접근했다가 틀렸다..

이 문제는 구슬의 개수가 최대 10개이므로 브루트포스로 접근을 해야 된다.

모든 경우의 수를 계산하면 해결할 수 있다.

소스 코드 보기

https://github.com/jokerKwu/BOJ_Algorithm/blob/master/python/BOJ_16198.py

jokerKwu/BOJ_Algorithm

import sys
N = sys.stdin.readline()
arr = list(map(int, sys.stdin.readline().split()))
answer = 0
def select(value):
    if len(arr) == 2:
        global answer
        if answer < value:
            answer = value
            return 0
    for i in range(1,len(arr)-1):
        value += (arr[i-1]*arr[i+1])
        tmp = arr[i]
        del arr[i]
        select(value)
        arr.insert(i,tmp)
        value -= (arr[i-1]*arr[i+1])

select(0)
print(answer)