[C++][BOJ/백준] 16198 에너지 모으기

 

 

Baekjoon Online Judge

 

클릭 시 문제로 이동

 

 

[문제]

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)

 

 

[출력]

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

 

 

---

 

[코드]

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int res = 0;

void dfs(vector<int>& v, int sum)
{
	if (v.size() == 2)
	{
		res = max(res, sum);

		return;
	}

	for (int i = 1; i < v.size() - 1; i++)
	{
		int tmp = v[i - 1] * v[i + 1];

		vector<int> copy = v;
		copy.erase(copy.begin() + i);

		dfs(copy, sum + tmp);
	}
}

int main()
{
	int N;

	cin >> N;

	vector<int> v;

	for (int i = 0; i < N; i++)
	{
		int w;

		cin >> w;
		v.push_back(w);
	}

	dfs(v, 0);

	cout << res << '\n';

	return 0;
}