17141 연구소 2

정민용·2023년 5월 29일
0

백준

목록 보기
247/286

문제

인체에 치명적인 바이러스를 연구하던 연구소에 승원이가 침입했고, 바이러스를 유출하려고 한다. 승원이는 연구소의 특정 위치에 바이러스 M개를 놓을 것이고, 승원이의 신호와 동시에 바이러스는 퍼지게 된다.

연구소는 크기가 N×N인 정사각형으로 나타낼 수 있으며, 정사각형은 1×1 크기의 정사각형으로 나누어져 있다. 연구소는 빈 칸, 벽으로 이루어져 있으며, 벽은 칸 하나를 가득 차지한다.

일부 빈 칸은 바이러스를 놓을 수 있는 칸이다. 바이러스는 상하좌우로 인접한 모든 빈 칸으로 동시에 복제되며, 1초가 걸린다.

연구소의 상태가 주어졌을 때, 모든 빈 칸에 바이러스를 퍼뜨리는 최소 시간을 구해보자.

# 17141
import sys
input = lambda: sys.stdin.readline().strip()

from collections import deque
from itertools import combinations

# 1. 벽 설치가 아닌 바이러스를 놓는 것
# 2. 바이러스를 놓을 수 있는 곳 중 m개는 바이러스를 놓고 나머지의 장소는 빈칸으로 설정한다
# 3. bfs 를 통해 최단시간을 구한다

n, m = map(int, input().split())
arr = [list(map(int, input().split())) for _ in range(n)]

virus = []
for x in range(n):
    for y in range(n):
        if arr[x][y] == 1:
            arr[x][y] = 10000
            continue
            
        elif arr[x][y] == 2:
            virus.append((x, y))
        arr[x][y] = -1
        
dx, dy = [-1, 0, 1, 0], [0, 1, 0, -1]


def bfs(virus_set):
    global n
    my_map = [[arr[x][y] for y in range(n)] for x in range(n)]
    
    queue = deque()
    for x, y in virus_set:
        queue.append((x, y))
        my_map[x][y] = 0
        
    while queue:
        x, y = queue.popleft()
        
        for i in range(4):
            nx, ny = x + dx[i], y + dy[i]
            if nx < 0 or ny < 0 or nx >= n or ny >= n:
                continue
            if my_map[nx][ny] == 10000:
                continue
                
            if my_map[nx][ny] == -1 or my_map[x][y] + 1 < my_map[nx][ny]:
                my_map[nx][ny] = my_map[x][y] + 1
                queue.append((nx, ny))
                
    time = 0
    for x in range(n):
        for y in range(n):
            if my_map[x][y] == 10000:
                continue
            if my_map[x][y] == -1:
                return 10000
            time = max(time, my_map[x][y])
    return time


min_time = 10000
for virus_set in combinations(virus, m):
    time = bfs(virus_set)
    min_time = min(min_time, time)
        
if min_time == 10000:
    print("-1")
else:
    print(min_time)

백준 17141 연구소 2

0개의 댓글