[Baekjoon] 에너지 모으기
Updated:
문제
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)
출력
첫째 줄에 모을 수 있는 에너지의 최댓값을 출력한다.
예제
Example 1:
Input:
4
1 2 3 4
Output:
12
Example 2:
Input:
5
100 2 1 3 100
Output:
10400
Example 3:
Input:
7
2 2 7 6 90 5 9
Output:
1818
Example 4:
Input:
10
1 1 1 1 1 1 1 1 1 1
Output:
8
조건
시간 제한 : 1초
메모리 제한 : 512 MB
풀이과정
풀이 1
import sys
sys.setrecursionlimit(10**9)
n = int(input())
goosle = list(map(int, input().split()))
ans = []
def dfs(goosle, energy):
if len(goosle) == 2:
ans.append(energy)
return
for i in range(len(goosle)):
if i != 0 and i != len(goosle)-1:
dfs(goosle[:i]+goosle[i+1:] ,energy+goosle[i-1]*goosle[i+1])
dfs(goosle, 0)
print(max(ans))
Leave a comment