인프런 커뮤니티 질문&답변

안녕하세요 선생님. 강의 잘 듣고 있습니다.

선생님의 풀이를 보니, 처음부터 모든 토마토가 익어있는 경우와 익은 토마토가 하나도 없는 경우는 안넣으신거같은데, 따로 이유가 있을까요?

또한, 아래는 제 풀이코드인데 dis 배열을 만들어서 선생님 풀이처럼 하는 것보다 복잡도가 높을까요?

import sys
from collections import deque
sys.stdin = open('in5.txt','r')

def total_search(arr,m,n):
    li = [0,0,0] # 토마토상태별로 갯수저장하는 갯수
    for i in range(n):
        for j in range(m):
            # 토마토가 없는 경우
            if arr[i][j] == -1:
                li[0] += 1
            # 토마토가 안익은 경우
            elif arr[i][j] == 0:
                li[1] += 1
            # 토마토가 익은 경우
            else:
                li[2] += 1
                dq.append((i,j,0))
    return li


if __name__ == "__main__":
    m, n = map(int,input().split()) # m = 가로, n = 세로
    arr = [list(map(int,input().split())) for _ in range(n)]  # 맵 정보
    dq = deque() # 익은 토마토 인덱스 저장 큐
    dy = [-1,0,1,0]
    dx = [0,1,0,-1]

    # 토마토 상태를 파악하는 함수
    li = total_search(arr,m,n)

    # 익은 토마토가 하나도 없는 경우
    if li[2] == 0:
        print(-1)
        sys.exit()
    # 처음부터 모든 토마토가 익어있을 경우
    elif li[2] == m*n :
        print(0)
        sys.exit()

    # 익는 과정 연산
    while dq:
        tmp = dq.popleft()
        y,x,d = tmp[0],tmp[1],tmp[2] # y,x 인덱스, d는 날짜

        # 상하좌우 연산
        for i in range(4):
            ny = y + dy[i]
            nx = x + dx[i]
            # 안익은 토마토일 경우
            if 0<=ny<=n-1 and 0<=nx<=m-1 and arr[ny][nx] == 0:
                # 익은 토마토로 바꾸고
                # 날짜 하나 올려서 큐에 추가
                arr[ny][nx] = 1
                dq.append((ny,nx,d+1))

    # 토마토 상태 갱신
    li = total_search(arr,m,n)

    # 토마토가 다 익지 못한 경우
    if li[1] != 0:
        print(-1)
    else:
        print(d)