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

dev.kkim님의 프로필 이미지
dev.kkim

작성한 질문수

파이썬 알고리즘 문제풀이 입문(코딩테스트 대비)

15. 토마토(BFS : Breadth First Search)

개수 문제

작성

·

175

0

import sys
from collections import deque
#sys.stdin = open("test.txt", "r")

m,n = map(int,input().split()) #m,n 순서
board = [list(map(int,input().split())) for _ in range(n)]

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

q = deque()
cnt = 0

for i in range(n):
    for j in range(m):
        if board[i][j] == 1:
            q.append((i,j))

if len(q) == 0:
    print(-1)
else:    
    while q:
        size = len(q) 
        for _ in range(size):
            temp = q.popleft()
            for k in range(4):
                x = temp[0] + dx[k]
                y = temp[1] + dy[k]
                if 0<=x<=n-1 and 0<=y<=m-1 and board[x][y] == 0:
                    q.append((x,y))
                    board[x][y] = 1
        cnt += 1

    flag = True
    for r in range(n):
        for c in range(m):
            if board[r][c] == 0:
                flag = False
                print(-1)
                break
        if flag == False:
            break
    if flag:
        print(cnt)
저는 큐의 길이만큼 for문을 돌면서 cnt를 하나씩 증가시킨 후 cnt를 출력했는데 정답이 하나씩 증가되서 나옵니다. 어느 부분이 논리적으로 잘못됐는지 찾지 못해서 질문드립니다.






답변 2

1

김태원님의 프로필 이미지
김태원
지식공유자

안녕하세요^^

상태트리의 레벨을 의미하는 cnt값이 마지막 레벨이 생기지도 않았는데 증가합니다.

if(len(q)): cnt += 1  이렇게 수정하면 될 것 같습니다.

0

dev.kkim님의 프로필 이미지
dev.kkim
질문자

감사합니다!!

dev.kkim님의 프로필 이미지
dev.kkim

작성한 질문수

질문하기