
N이 5, K가 4라는 점에서 경우의 수가 적음을 예측할 수 있습니다.
K가 8이더라도 5^8은 그리 큰 수가 아닙니다.
#include <iostream>
using namespace std;
int N, K, answer;
int A[5], RM[8][5];
void dfs(int day, int sum)
{
if (day == K * 2)
{
answer = max(sum, answer);
return;
}
for (int i = 0; i < N; ++i)
{
if (!A[i])
{
continue;
}
--A[i];
dfs(day + 1, sum + RM[day][i]);
++A[i];
}
}
int main()
{
ios::sync_with_stdio(0), cin.tie(0);
cin >> N >> K;
for (int i = 0; i < N; ++i)
{
cin >> A[i];
}
for (int i = 0; i < K * 2; ++i)
{
for (int j = 0; j < N; ++j)
{
cin >> RM[i][j];
}
}
dfs(0, 0);
cout << answer;
return 0;
}
경우의 수가 적다는 점에서 모든 경우의 수를 확인하는 방법으로 합의 최댓값을 구할 수 있습니다.
즉 DFS와 같은 방법으로 모든 경우에 대해 방문하는 형식으로 해결할 수 있습니다.