백준 16198 c++ 에너지 모으기

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

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)

출력

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

---

#include <iostream>
using namespace std;

int N;//횟수
int arr[10];//숫자 입력받을 배열
int check[10];//기준점으로 지정했을 때 true로
int sum[8];//한번의 재귀에서 합들을 저장할 배열
int MAX = 0;//합의 최대값

void dfs(int count)
{
	if (count == N - 2)
	{
		int result = 0;
		for (int i = 0; i < 8; i++)
		{
			result += sum[i];//모든 합 더해주기
		}
		MAX = max(MAX, result);//기존 max값과 비교
		return;
	}

	for (int i = 1; i < N - 1; i++)
	{
		if (check[i])//이미 체크한 기준점이면 continue
			continue;
		check[i] = true;//현재 기준점을 true로
		int front = i - 1;
		int back = i + 1;
		while (check[front])//true 기준점으로 사용되지 않은 앞까지 땡긴다.
			front--;
		while (check[back])//반대로 뒤로 땡김
			back++;
		sum[count] = arr[front] * arr[back];
		dfs(count + 1);
		check[i] = false;
	}

}

int main()
{
	cin >> N;
	for (int i = 0; i < N; i++)
	{
		cin >> arr[i];
	}

	dfs(0);
	cout << MAX;

}