[백준]18405번: 경쟁적 전염

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

18405번: 경쟁적 전염

---

처음에 짠 코드에서 시간 카운트하는 로직을 잘못 짜서 자꾸 오답 판정이 나왔었다.

너무 오래 붙잡고 있는 것 같아서 정답 예시 코드를 봤다. (이코테 책)

큐에 좌표데이터를 넣을 때 좌표와 함께 시간정보까지 같이 담으면 제대로 카운트 할 수 있었다.

 

from collections import deque
import sys
input = sys.stdin.readline

dx=[-1,1,0,0]
dy=[0,0,1,-1]

def bfs(q,time): 
  while q:
    virus, s, x, y = q.popleft()
    if s==time:
      return
    for i in range(4):
      nx, ny = x+dx[i], y+dy[i]
      if 0<=nx<n and 0<=ny<n:
        if graph[nx][ny]==0:
          graph[nx][ny] = virus
          q.append((virus, s+1, nx, ny))
  return
  

n,k = map(int,input().split())
graph=[]
for i in range(n):
  graph.append(list(map(int,input().split())))

data=[] #바이러스 정보
for i in range(n):
  for j in range(n):
    if graph[i][j]!=0:
      #바이러스 종류, 시간정보, 좌표정보
      data.append((graph[i][j], 0, i, j))

data.sort() #번호가 낮은 종류의 바이러스부터 먼저 증식하므로 정렬해줌
queue = deque(data) #큐에 정렬된 바이러스 정보 삽입

s,a,b=map(int,input().split())

bfs(queue,s)
print(graph[a-1][b-1])