240617 [BOJ/백준] 16198. 에너지 모으기

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

---

문제

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)

---

출력

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

---

입력 코드

def energy():
    global ans, sum_v
    if len(weight) == 2:
        ans = max(ans, sum_v)
        return

    for i in range(1, len(weight)-1):
        sum_v = sum_v + weight[i-1] * weight[i+1]
        x = weight.pop(i)
        energy()
        weight.insert(i, x)
        sum_v = sum_v - weight[i-1] * weight[i+1]

n = int(input())
weight = list(map(int, input().split()))
ans = 0
sum_v = 0

energy()
print(ans)

---

결과

---

이건 백트래킹에 자신없어서 찾아보다가 나중에 도전해봐야겠다고 생각한 문제였다.

이 문제를 풀면서 겪은 문제는 2개가 있다.

1. index에러: 그 이유는 선택한 구슬 x를 먼저 제거하고 양 옆의 에너지의 곱을 더했기 때문이다. (입력 코드의 1번째 줄과 2번째 줄의 순서를 바꿈)

2. 제대로 된 값을 계산하지 못함:  x = weight.pop(i)의 코드를 weight.remove(weight[i])로 썼기 때문이다. 나는 weight[i]값이 변하지 않을거라고 생각했지만 i값은 그대로 있더라도 remove로 지워진 weight[i]값은 저장하지 않으면 찾을 수 없어 남아있는 weight에서 인덱스 i 값을 insert하는 것처럼 보였다.

두 가지 문제를 해결하고 성공! 그래도 전보다는 백트래킹에 대한 자신감이 조금은 생긴 것 같다!ㅎㅎ