당신이 다니는 학교는 매년 체육대회를 합니다. 체육대회는 여러 종목에 대해 각 반의 해당 종목 대표가 1명씩 나와 대결을 하며, 한 학생은 최대 한개의 종목 대표만 할 수 있습니다. 당신의 반에서도 한 종목당 1명의 대표를 뽑으려고 합니다. 학생들마다 각 종목에 대한 능력이 다르지만 이 능력은 수치화되어 있어 미리 알 수 있습니다. 당신의 반의 전략은 각 종목 대표의 해당 종목에 대한 능력치의 합을 최대화하는 것입니다.
다음은 당신의 반 학생이 5명이고, 종목의 개수가 3개이며, 각 종목에 대한 학생들의 능력치가 아래 표와 같을 때, 각 종목의 대표를 뽑는 예시입니다.
이름 | 테니스 | 탁구 | 수영 |
---|---|---|---|
석환 | 40 | 10 | 10 |
영재 | 20 | 5 | 0 |
인용 | 30 | 30 | 30 |
정현 | 70 | 0 | 70 |
준모 | 100 | 100 | 100 |
테니스 대표로 준모, 탁구 대표로 인용, 수영 대표로 정현을 뽑는다면, 세 명의 각 종목에 대한 능력치의 합은 200(=100+30+70)이 됩니다.
하지만, 테니스 대표로 석환, 탁구 대표로 준모, 수영 대표로 정현을 뽑는다면 세 명의 각 종목에 대한 능력치 합은 210(=40+100+70)이 됩니다. 이 경우가 당신의 반의 각 종목 대표의 능력치 합이 최대가 되는 경우입니다.
당신의 반 학생들의 각 종목에 대한 능력치를 나타내는 2차원 정수 배열 ability가 주어졌을 때, 선발된 대표들의 해당 종목에 대한 능력치 합의 최대값을 return 하는 solution 함수를 완성하시오.
ability | result |
---|---|
[[40, 10, 10], [20, 5, 0], [30, 30, 30], [70, 0, 70], [100, 100, 100]] | 210 |
[[20, 30], [30, 20], [20, 30]] | 60 |
class Solution {
boolean[] VISIT = new boolean[10];
int[] MATRIX = new int[10];
int maxValue = -1;
public int solution(int[][] ability) {
DFS(0, ability[0].length, ability.length, ability);
return maxValue;
}
public void DFS(int count, int n, int max, int[][] ability) {
if(count == n){
int result = 0;
for(int i =0; i < n; i++){
result += ability[MATRIX[i]][i];
}
maxValue = Math.max(maxValue, result);
return;
}
for(int i = 0; i < max; i++){
if(!VISIT[i]){
VISIT[i] = true;
MATRIX[count] = i;
DFS(count+1, n, max, ability);
VISIT[i] = false;
}
}
}
}
탐색해야 하는 경우가 그리 많지 않아서 완전 탐색으로 때워서 풀은 문제.
하지만 완전탐색은 데이터 양이 적은 상황일 때 써먹을 만 하고 DP를 사용하는 게 좀 더 좋은 방법.