[BOJ] 17135 캐슬 디펜스

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

17135번: 캐슬 디펜스

문제

캐슬 디펜스는 성을 향해 몰려오는 적을 잡는 턴 방식의 게임이다. 게임이 진행되는 곳은 크기가 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

풀이

나를 겁쟁이라고 할수도 있겠지만.. 나는 이런 시뮬레이션 구현 문제가 좋다. 뭔가 실제 프로그램 만드는 기분?이 들어서 재미있게 풀 수 있는 것 같다. 이번 문제 또한 시뮬레이션 문제이고, N, M이 비교적 작으므로 브루트포스로 풀 수 있을 것 같았다. 문제에서 주어진 조건을 그대로 코드로 구현하기만 하는 문제였기에 크게 어렵진 않았다. 먼저 궁수들의 위치할 수 있는 모든 경우의 수를 combinations로 구하였다. 각 경우의 수를 스캔하면서 각 위치에서 가장 가까운 적을 구하는 방법은 우선순위 큐 원리를 이용해서 구하였다. tmp_defense라는 defense배열을 deepcopy한 배열에서 만약 적이 없다면(enemy_present == False) 멈추고 그 경우의 수에 대하여 처치한 적의 수가  max_cnt보다 크면 max_cnt에 cnt를 대입해주었다.

import sys
from itertools import combinations
input = lambda: sys.stdin.readline().rstrip()
n, m, d = map(int, input().split())
defense = [list(map(int, input().split()))for _ in range(n)]
pos = list(combinations([i for i in range(m)], 3))
max_cnt = 0
for p in pos:
  cnt = 0
  tmp_defense = [[defense[i][j] for j in range(m)]for i in range(n)]
  while True:
    target = []
    for k in range(len(p)):
      enemies = []
      for i in range(n):
        for j in range(m):
          if tmp_defense[i][j] == 1:
            if n-i + abs(p[k]-j) <= d:
              enemies.append([n-i+abs(p[k]-j), j, i])
      enemies.sort()
      if enemies:
        target.append([enemies[0][2], enemies[0][1]])
    if target:
      for t in target:
        if tmp_defense[t[0]][t[1]] == 1:
          tmp_defense[t[0]][t[1]] = 0
          cnt += 1
    enemy_present = False
    for i in range(n-1, -1, -1):
      for j in range(m):
        if tmp_defense[i][j] == 1:
          if i+1 == n:
            tmp_defense[i][j] = 0
            continue
          enemy_present = True
          tmp_defense[i+1][j] = 1
          tmp_defense[i][j] = 0
    if not enemy_present:
      if max_cnt < cnt:
        max_cnt = cnt
      break
print(max_cnt)