[백준] 16198번 에너지 모으기 - 파이썬(Python)

문제

N개의 에너지 구슬이 일렬로 놓여져 있고, 에너지 구슬을 이용해서 에너지를 모으려고 한다.

i번째 에너지 구슬의 무게는 Wi이고, 에너지를 모으는 방법은 다음과 같으며, 반복해서 사용할 수 있다.

1. 에너지 구슬 하나를 고른다. 고른 에너지 구슬의 번호를 x라고 한다. 단, 첫 번째와 마지막 에너지 구슬은 고를 수 없다.
2. x번째 에너지 구슬을 제거한다.
3. Wx-1 × Wx+1의 에너지를 모을 수 있다.
4. N을 1 감소시키고, 에너지 구슬을 1번부터 N번까지로 다시 번호를 매긴다. 번호는 첫 구슬이 1번, 다음 구슬이 2번, ... 과 같이 매겨야 한다.

N과 에너지 구슬의 무게가 주어졌을 때, 모을 수 있는 에너지 양의 최댓값을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 에너지 구슬의 개수 N(3 ≤ N ≤ 10)이 주어진다.

둘째 줄에는 에너지 구슬의 무게 W1, W2, ..., WN을 공백으로 구분해 주어진다. (1 ≤ Wi ≤ 1,000)

출력

첫째 줄에 모을 수 있는 에너지의 최댓값을 출력한다.

문제 링크: https://www.acmicpc.net/problem/16198

16198번: 에너지 모으기

    n = int(input())
    arr = list(map(int, input().split()))
    answer = 0

    def dfs(li, total):
        global answer
        if len(li) == 2:
            answer = max(answer,total)
            return
        for i in range(1, len(li)-1):
            dfs(li[:i] + li[i+1:], total + (li[i-1] * li[i+1]))


    dfs(arr, answer)
    print(answer)

list는 골라진 요소를 삭제하면 나머지 요소들이 빈 자리를 메꾸기 위해 이동하므로 시간이 많이 걸립니다. DFS에 골라진 요소를 제외한 앞부분의 list, 뒷부분의 list를 합한 리스트를 보내면 해결이 됩니다. 맨 앞과 맨 뒤만 남았을때 종료하면 되므로 len(li) == 2일때 종료를 해줍니다.