17135. 캐슬 디펜스

업데이트 시간 : 2023-03-17 06:44:35 +0000

[Gold III] 캐슬 디펜스 - 17135

성능 요약

메모리: 116236 KB, 시간: 212 ms

분류

구현, 그래프 이론, 브루트포스 알고리즘, 그래프 탐색, 너비 우선 탐색, 시뮬레이션

문제 설명

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

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

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

격자판의 두 위치 (r₁, c₁), (r₂, c₂)의 거리는 |r₁-r₂| + |c₁-c₂|이다.

입력

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

출력

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

💡 Solutions

📄 캐슬 디펜스.py

N,M,D = map(int,input().split())

matrix = [list(map(int,input().split())) for _ in range(N)] # 게임판

animys = []
def is_finish(n):
  '''
  게임을 시작할 때, 적군을 탐색한다.
  # 이 때, 왼쪽부터 먼저 잡는다는 조건이 있기 때문에 열방향으로 우선 탐색하는 것이 아니라
  # 행방향으로 우선적으로 탐색을 하여 정렬을 한다.
  '''
  global animys
  animys = []
  Trigger = True
  for j in range(M):           
    for i in range(n-1,-1,-1): 
      if play[i][j] == 1:
        animys.append((i,j))
        Trigger = False
  return True if Trigger else False

from itertools import combinations # 이거도 조합을 사용할 예정 

new_line = [[0] * M] # 한턴이 끝날 때 마다 게임이 끝나는 걸 표시 // 탐색을 할 때 range를 계속 헤아리는 것 보다 좋다고 판단했었음
result = 0
for bows in combinations(range(M), 3): # 궁수의 자리 배치 결정
  play = [arr[:] for arr in matrix]    # 탐색을 위해 deepcopy 진행
  n = N
  kill = 0   # 총 몇명의 적을 죽였나
  while not is_finish(n):
    
    targets = set() # 화살을 하나의 적이 같이 맞을 수 있다고 했으므로, set을 이용하여 적을 헤아린다.

    for bow in bows: # 모든 궁수에 대해서 적을 탐색한다.
      
      _min = n+M
      target = None
      
      for x,y in animys:
        length = abs(n-x) + abs(bow-y)
        if length <= D:
          if length < _min:           # 왼쪽부터 카운팅을 하기 때문에 중복은 고려하면 안된다.
            _min = length
            target = (x,y)

      if target : targets.add(target) # 잡을 녀석이 있으면 set에 더한다. 
    kill += len(targets) # target 길이만큼 적을 제거
    
    # 타겟들은 죽었습니다..
    for target in targets:
      if not target : continue
      play[target[0]][target[1]] = 0
    
    play.pop()
    n -= 1
  
  # result 중 큰 값을
  result = result if result > kill else kill

print(result)