백준 17135번 캐슬 디펜스 // JAVA

문제

캐슬 디펜스

캐슬 디펜스는 성을 향해 몰려오는 적을 잡는 턴 방식의 게임이다. 게임이 진행되는 곳은 크기가 N×M인 격자판으로 나타낼 수 있다. 격자판은 1×1 크기의 칸으로 나누어져 있고, 각 칸에 포함된 적의 수는 최대 하나이다. 격자판의 N번행의 바로 아래(N+1번 행)의 모든 칸에는 성이 있다.

 

성을 적에게서 지키기 위해 궁수 3명을 배치하려고 한다. 궁수는 성이 있는 칸에 배치할 수 있고, 하나의 칸에는 최대 1명의 궁수만 있을 수 있다. 각각의 턴마다 궁수는 적 하나를 공격할 수 있고, 모든 궁수는 동시에 공격한다. 궁수가 공격하는 적은 거리가 D이하인 적 중에서 가장 가까운 적이고, 그러한 적이 여럿일 경우에는 가장 왼쪽에 있는 적을 공격한다. 같은 적이 여러 궁수에게 공격당할 수 있다. 공격받은 적은 게임에서 제외된다. 궁수의 공격이 끝나면, 적이 이동한다. 적은 아래로 한 칸 이동하며, 성이 있는 칸으로 이동한 경우에는 게임에서 제외된다. 모든 적이 격자판에서 제외되면 게임이 끝난다.

 

게임 설명에서 보다시피 궁수를 배치한 이후의 게임 진행은 정해져있다. 따라서, 이 게임은 궁수의 위치가 중요하다. 격자판의 상태가 주어졌을 때, 궁수의 공격으로 제거할 수 있는 적의 최대 수를 계산해보자.

격자판의 두 위치 (r1, c1), (r2, c2)의 거리는 |r1-r2| + |c1-c2|이다.

입력

첫째 줄에 격자판 행의 수 N, 열의 수 M, 궁수의 공격 거리 제한 D가 주어진다. 둘째 줄부터 N개의 줄에는 격자판의 상태가 주어진다. 0은 빈 칸, 1은 적이 있는 칸이다.

3 ≤ N, M ≤ 15
1 ≤ D ≤ 10

출력

첫째 줄에 궁수의 공격으로 제거할 수 있는 적의 최대 수를 출력한다.

---

풀이

DFS를 활용한 조합을 통해 궁수가 위치할 수 있는 경우의 수를 모두 탐색하였다.

궁수의 위치 3가지가 정해지면, 처치할 수 있는 적의 수를 세고 그 값의 최대치를 구하면 된다.

조심해야 할 점은, 문제의 조건인데

1. 가장 가까이 있는 적을 처치
2. 거리가 같다면 가장 왼쪽에 있는 적을 처치
3. 각각의 궁수가 같은 적을 처치할 수도 있음

이 조건만 유의하면 쉽게 해결할 수 있다.

---

코드

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
	static int N, M, D;
	static int[][] castle;
	static int[] archer;
	static int answer;

	static class Pos {
		int x, y;

		Pos() {
		}

		Pos(int x, int y) {
			this.x = x;
			this.y = y;
		}
	}

	static Pos findMinDist(int[][] temp, int bow) {
		Pos pos = new Pos(-1, -1);
		int min = Integer.MAX_VALUE;

		for (int i = 0; i < M; i++) {
			for (int j = N - 1; j >= 0; j--) {
				if(temp[j][i] == 1) {
					int dist = Math.abs(bow - i) + Math.abs(N - j);
					if(min > dist && dist<= D) {
						min = dist;
						pos.x = i;
						pos.y = j;
					}
				}
			}
		}

		return pos;
	}

	static void dfs(int index, int count, boolean[] visit) {
		if (count >= 3) {
			int sum = 0;
			int[][] temp = new int[N][M];

			for (int i = 0; i < N; i++) {
				System.arraycopy(castle[i], 0, temp[i], 0, M);
			}

			for (int i = 0; i < N; i++) { // N 번만큼 내려옴
				Pos[] die = new Pos[3];

				for (int j = 0; j < 3; j++) {
					die[j] = findMinDist(temp, archer[j]);
				}

				for (int j = 0; j < 3; j++) {
					Pos cur = die[j];

					if (cur.x == -1)
						continue;

					if (temp[cur.y][cur.x] == 1) {
						temp[cur.y][cur.x] = 0;
						sum++;
					}
				}

				for (int j = N - 1; j > 0; j--) {
					for (int k = 0; k < M; k++) {
						temp[j][k] = temp[j - 1][k];
					}
				}

				for (int j = 0; j < M; j++) {
					temp[0][j] = 0;
				}
			}

			answer = Math.max(sum, answer);

			return;
		}

		for (int i = index; i < M; i++) {
			if (!visit[i]) {
				visit[i] = true;
				archer[count] = i;
				dfs(i + 1, count + 1, visit);
				visit[i] = false;
			}
		}
	}

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());

		N = Integer.parseInt(st.nextToken());
		M = Integer.parseInt(st.nextToken());
		D = Integer.parseInt(st.nextToken());

		castle = new int[N][M];
		archer = new int[3];
		answer = 0;

		for (int i = 0; i < N; i++) {
			st = new StringTokenizer(br.readLine());

			for (int j = 0; j < M; j++) {
				castle[i][j] = Integer.parseInt(st.nextToken());
			}
		}

		dfs(0, 0, new boolean[M]);

		System.out.println(answer);
	}
}