https://www.acmicpc.net/problem/3980
3980번: 선발 명단
각각의 테스트 케이스에 대해서, 모든 포지션의 선수를 채웠을 때, 능력치의 합의 최댓값을 한 줄에 하나씩 출력한다. 항상 하나 이상의 올바른 라인업을 만들 수 있다.
www.acmicpc.net
코드설명
DFS를 활용한 브루트포스(깊이우선탐색, bruteforce) 문제입니다.
모든 경우를 완전탐색해줍니다.
유의해야할점은 다음과 같습니다.
1. 선수가 해당 포지션 능력치가 0 이라면 들어가지 못하게 처리합니다. 즉, arr[level][i] > 0 일 경우에만 탐색하도록 처리합니다.
2. 그 포지션을 다른 선수가 사용하기로했다면, 다른 선수는 사용하지 못합니다.
코드
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
public class Main {
public static int N, T;
public static int[][] map;
public static int[][] arr;
public static boolean[] positionVisited;
public static int cnt = 0;
public static int answer = 0;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
T = Integer.parseInt(st.nextToken());
for(int t=0; t<T;t++) {
answer = 0;
arr = new int[11][11];
positionVisited = new boolean[11];
for(int i=0;i<11;i++) {
st = new StringTokenizer(br.readLine());
for(int j=0;j<11;j++) {
arr[i][j] = Integer.parseInt(st.nextToken());
}
}
dfs(0, 0);
System.out.println(answer);
}
}
public static void dfs(int level, int sum) {
if(level == 11) {
answer = Math.max(answer, sum);
return ;
}
for(int i=0;i<11;i++) {
if(arr[level][i] > 0 && positionVisited[i] == false) {
int temp = arr[level][i];
positionVisited[i] = true;
arr[level][i] = 0;
dfs(level + 1, sum + temp);
positionVisited[i] = false;
arr[level][i] = temp;
}
}
}
}
처음에는 visited[level]. 즉, 해당 선수의 레벨까지 확인했었는데 생각해보면 이전 Level을 다시 체크할 일은 없기에 사용하지않는다.
아래코드보다는 위의 코드로 사용.
public static void dfs(int level, int sum) {
if(level == 11) {
// System.out.println("sum:"+sum);
answer = Math.max(answer, sum);
return ;
}
for(int i=0;i<11;i++) {
if(visited[level] == false && arr[level][i] > 0 && positionVisited[i] == false) {
int temp = arr[level][i];
visited[level] = true;
positionVisited[i] = true;
arr[level][i] = 0;
dfs(level + 1, sum + temp);
visited[level] = false;
positionVisited[i] = false;
arr[level][i] = temp;
}
}
}