[백준 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명의 궁수를 배치시킬 수 있는 경우를 구하는 것으로 시작한다.

궁수의 위치를 구한 후에 맵에 적이 없어질 때까지 매 턴마다 적을 죽이고 적을 이동시킨다.

적이 모두 없어졌다면 죽인 적의 수를 통해 최댓값을 구한다.

 

어떻게 적을 죽이고, 이동하는지 등을 잘 구현해야하고 코드를 좀 더 효율적으로 구현하는 연습을 해야겠다.

 

구현

- 먼저 3명의 궁수를 배치시킨다.

- 3명의 궁수를 배치시켰다면 궁수의 위치를 담은 리스트를 kill 함수에 전달한다.

def combi(next, picked):
    if next >= M:
        return

    if len(picked) == 3:
        kill(picked)
        return

    for i in range(next, M):
        if not visited[i]:
            picked.append(i)
            visited[i] = True
            combi(i, picked)
            picked.pop()
            visited[i] = False

- 먼저 원본 맵을 복사하여 temp_map에 저장한다.

- flag가 True일 때 계속 턴이 진행되고 맵에 1이 없을 때 flag는 False가 유지된 채로 턴이 종료된다.

- 먼저 궁수들의 위치를 기준으로 각각 죽여야하는 적의 위치를 찾는다.

- 적의 위치가 D이하일 때 조건에 맞는 적을 찾아 target_list에 위치를 저장한다.

- 이때 궁수들은 동시에 화살을 쏘기 때문에 먼저 각자 쏘려고하는 적을 찾고 나중에 동시에 죽여야한다.

- 궁수들이 화살을 쏠 적을 찾았다면 적들을 맵에서 제거한다.

- 해당 위치가 1일 때 0을 대입하고 죽인 적들의 수인 count를 증가시킨다. 여기서 해당 맵이 이미 0이라면 다른 궁수가 이미 죽인 것이다.

- 턴이 종료되면 최댓값을 비교하여 갱신한다.

def kill(pos_list):
    global answer
    temp_map = [[0 for _ in range(M)] for _ in range(N)]
    for i in range(N):
        for j in range(M):
            temp_map[i][j] = board[i][j]

    count = 0
    flag = True
    while flag:
        flag = False
        target_list = []
        for pos in pos_list:
            killer_y = N
            killer_x = pos
            target_y, target_x = -1, -1
            for y in range(N-1, -1, -1):
                for x in range(M):
                    if temp_map[y][x] == 0:
                        continue
                    if distance(killer_y, killer_x, y, x) > D:
                        continue
                    else:
                        if target_y == -1:
                            target_y, target_x = y, x
                        else:
                            if distance(killer_y, killer_x, target_y, target_x) > distance(killer_y, killer_x, y, x):
                                target_y, target_x = y, x
                            elif distance(killer_y, killer_x, target_y, target_x) == distance(killer_y, killer_x, y, x):
                                if target_x > x:
                                    target_y, target_x = y, x
            if target_y != -1:
                target_list.append((target_y, target_x))

        for y, x in target_list:
            if temp_map[y][x] == 1:
                temp_map[y][x] = 0
                count += 1

        for x in range(M):
            for y in range(N-1, -1, -1):
                if temp_map[y][x] == 1:
                    flag = True
                    if y == N-1:
                        temp_map[y][x] = 0
                    else:
                        temp_map[y][x] = 0
                        temp_map[y+1][x] = 1

    if answer < count:
        answer = count

---

전체 코드

import sys
input = sys.stdin.readline

def distance(cy, cx, ny, nx):
    return abs(cy-ny) + abs(cx-nx)

def kill(pos_list):
    global answer
    temp_map = [[0 for _ in range(M)] for _ in range(N)]
    for i in range(N):
        for j in range(M):
            temp_map[i][j] = board[i][j]

    count = 0
    flag = True
    while flag:
        flag = False
        target_list = []
        for pos in pos_list:
            killer_y = N
            killer_x = pos
            target_y, target_x = -1, -1
            for y in range(N-1, -1, -1):
                for x in range(M):
                    if temp_map[y][x] == 0:
                        continue
                    if distance(killer_y, killer_x, y, x) > D:
                        continue
                    else:
                        if target_y == -1:
                            target_y, target_x = y, x
                        else:
                            if distance(killer_y, killer_x, target_y, target_x) > distance(killer_y, killer_x, y, x):
                                target_y, target_x = y, x
                            elif distance(killer_y, killer_x, target_y, target_x) == distance(killer_y, killer_x, y, x):
                                if target_x > x:
                                    target_y, target_x = y, x
            if target_y != -1:
                target_list.append((target_y, target_x))

        for y, x in target_list:
            if temp_map[y][x] == 1:
                temp_map[y][x] = 0
                count += 1

        for x in range(M):
            for y in range(N-1, -1, -1):
                if temp_map[y][x] == 1:
                    flag = True
                    if y == N-1:
                        temp_map[y][x] = 0
                    else:
                        temp_map[y][x] = 0
                        temp_map[y+1][x] = 1

    if answer < count:
        answer = count


def combi(next, picked):
    if next >= M:
        return

    if len(picked) == 3:
        kill(picked)
        return

    for i in range(next, M):
        if not visited[i]:
            picked.append(i)
            visited[i] = True
            combi(i, picked)
            picked.pop()
            visited[i] = False

N, M, D = map(int, input().split())
board= [list(map(int, input().split())) for _ in range(N)]

answer = 0

visited = [False] * M
for m in range(M):
    visited[m] = True
    combi(m, [m,])
    visited[m] = False

print(answer)