[BOJ] 에너지 모으기 (no.16198)
문제
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)
출력
첫째 줄에 모을 수 있는 에너지의 최댓값을 출력한다.
🤔 생각
-
쌩순열로 풀어야 하나 생각했지만 시간제한이 1초라 위험할 것 같았다.
-
일단 재귀함수로 구현해보자.
📌 1차 풀이
import sys
input = sys.stdin.readline
#print = sys.stdout.write
def func(pre):
global ans
if len(stack) <= 2:
ans = max(ans, pre)
return
else:
for i in range(1, len(stack) - 1):
res = stack[i-1]*stack[i+1]
node = stack[i]
del stack[i]
func(pre + res)
stack.insert(i, node)
n = int(input())
stack = list(map(int, input().split()))
ans = 0
func(0)
print(ans)
-
그렇게 확신을 가지고 짠 코드는 아닌데 한번에 통과해서 당황+신남
-
리스트 명이 스택인건.. 스택 DFS로 해보려다가 망해서 그런거니 넘어가주길 바란다.
-
다른 사람의 풀이를 보며 현재 코드를 최대한 개선해보았다.
- 리스트 삽입 삭제 말고, 슬라이싱만 해서 파라미터로 넘겨주기
- 리스트를 변경할 일이 없어지므로, 튜플로 정의하여 메모리를 절약하자.
📌 개선 풀이
import sys
input = sys.stdin.readline
print = sys.stdout.write
n = int(input())
w = tuple(map(int, input().split()))
def dfs(arr):
if len(arr) == 3:
return arr[0]*arr[2]
ans = 0
for i in range(1, len(arr)-1):
res = arr[i-1]*arr[i+1] + dfs(arr[:i] + arr[i+1:])
ans = max(res, ans)
return ans
print("%d" % dfs(w))
Python3 기준 132ms -> 108ms 로 줄었다.
✔ 회고
- 불변 데이터라면 튜플로 선언하는 습관을 들이자.
Author And Source
이 문제에 관하여([BOJ] 에너지 모으기 (no.16198)), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@wisepine/BOJ-에너지-모으기-no.16198저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)