이호진
백준 16198 파이썬 - 에너지 모으기 - 2021/05/23 본문
문제 설명
N개의 에너지 구슬이 일렬로 놓여져 있고, 에너지 구슬을 이용해서 에너지를 모으려고 한다.
i번째 에너지 구슬의 무게는 Wi이고, 에너지를 모으는 방법은 다음과 같으며, 반복해서 사용할 수 있다.
- 에너지 구슬 하나를 고른다. 고른 에너지 구슬의 번호를 x라고 한다. 단, 첫 번째와 마지막 에너지 구슬은 고를 수 없다.
- x번째 에너지 구슬을 제거한다.
- Wx-1 × Wx+1의 에너지를 모을 수 있다.
- N을 1 감소시키고, 에너지 구슬을 1번부터 N번까지로 다시 번호를 매긴다. 번호는 첫 구슬이 1번, 다음 구슬이 2번, ... 과 같이 매겨야 한다.
N과 에너지 구슬의 무게가 주어졌을 때, 모을 수 있는 에너지 양의 최댓값을 구하는 프로그램을 작성하시오.
첫째 줄에 에너지 구슬의 개수 N(3 ≤ N ≤ 10)이 주어진다.
둘째 줄에는 에너지 구슬의 무게 W1, W2, ..., WN을 공백으로 구분해 주어진다. (1 ≤ Wi ≤ 1,000)
첫째 줄에 모을 수 있는 에너지의 최댓값을 출력한다.
풀이
DFS를 이용해서 모든 경우의 수를 구한다.
n = int(input())
arr = list(map(int, input().split()))
result = []
def dfs(value):
if len(arr) == 2:
result.append(value)
return
else:
for i in range(1, len(arr) - 1):
gop = arr[i - 1] * arr[i + 1]
temp = arr[i]
del arr[i]
dfs(value + gop)
arr.insert(i, temp)
dfs(0)
print(max(result))
'백준문제풀이' 카테고리의 다른 글
백준 11724 파이썬 - 연결 요소의 개수 - 2021/05/23 (0) | 2021.05.23 |
---|---|
백준 2178 파이썬 - 미로 탐색 - 2021/05/22 (0) | 2021.05.22 |
백준 14889 파이썬 - 스타트와 링크 - 2021/05/21 (0) | 2021.05.21 |
백준 10819 파이썬 - 차이를 최대로 - 2021/05/21 (0) | 2021.05.21 |
백준 2606 파이썬 - 바이러스 - 2021/05/18 (0) | 2021.05.18 |