2-seongwon030

[seongwon030]

🔗 문제 링크

토마토

✔️ 소요된 시간

1일

✨ 수도 코드

1. 토마토가 1인 좌표부터 동시에 시작한다.
2. 좌표의 크기를 넘기면 안된다.
3. 큐에 좌표를 넣으면서 다음으로 계속 넘어가기
4. 큐가 빌 때까지 반복

첫번째 풀이

from collections import deque

m,n = map(int,input().split())
tomato = [list(map(int,input().split())) for _ in range(n)]
queue = deque([]) # 좌표 -> 리스트

dx, dy = [-1,1,0,0], [0,0,-1,1] # 좌,우,하,상 -> 좌표 이동 
ans = 0 # 최소 날짜 

for i in range(n):
  for j in range(m):
    if tomato[i][j] == 1: # 1인 경우 토마토가 있으므로 큐에 append
      queue.append([i,j])

def bfs():
  while queue:
    x,y = queue.popleft() # 첫 번째 좌표 
    for i in range(4):
      nx,ny = dx[i] + x, dy[i] + y
      # 좌표 크기를 넘어가선 안되고 값이 0이어야 
      if 0 <= nx < n and 0 <= ny < m and tomato[nx][ny] == 0:
        tomato[nx][ny] = tomato[x][y] + 1 # 그 다음 칸으로 이동
        queue.append([nx,ny])

bfs()
for i in tomato:
  for j in i:
    if j == 0:
      print(-1)
      exit(0)
    ans = max(ans, max(i))
print(ans-1)

1. 토마토의 상태를 나타내는 위치를 2차원배열로 저장
2. 진행상태를 알려주는 덱을 리스트로 설정
3. 좌표를 상하좌우로 나눈 리스트 설정
4. 값이 1인경우 큐에 append
5. 좌표를 꺼내어 x,y좌표를 nx,ny만큼 계산한 결과가 좌표크기를 넘어가선 안되며 0일때만 토마토의 상태가 0에서 1로 변함.
6. 해당 좌표를 큐에 다시 append
7. 5와 6을 큐가 비어있을 때까지 반복
8. 토마토 상태를 나타내는 리스트를 반복문으로 돌면서 0인 값이 있으면 -1을 출력하고 exit.
9. 없으면 tomato 를 순회하며 각 칸의 최댓값을 찾고 (max(i)는 한 행의 최댓값), 모든 행의 최댓값 중 가장 큰 값을 ans에 저장.
10. 익은 토마토의 값은 1로 시작하므로 걸리는 일수는 1을 뺀 ans-1

문제점

=> -1일 때 계속 출력하는 문제

두번째 풀이

flag = False # 안 익은 토마토 있는지

for i in tomato:
  if 0 in i:
    flag = True
    break

if flag: # 안 익은 토마토가 존재하므로 -1출력
  print(-1)
else:
  for i in tomato:
    ans = max(ans,max(i))
  print(ans-1)

1. 안 익은 토마토의 유무를 알기 위해 flag 에 불리언 값인 false로 저장
2. tomato 리스트를 순회하며 0이 있는 경우엔 flag를 True로 바꾸고 break
3. flag가 True면 안 익은 토마토가 있으므로 -1을 출력
4. 아니면 첫 번째 풀이에서와 같이 ans 출력

📚 새롭게 알게된 내용

x와 y좌표를 모두 사용해서 이동하는 경우는 처음 접해봤고 2차원 리스트를 이용해 풀 수 있는게 참신했다.

[InSange]

저도 토마토 참 좋아하는데요?
만약 좀 더 시간복잡도를 신경쓰고 싶다면 입력 받을 때 덜 익은 토마토0 개수를 cnt 로 따로 분류하여 익을 때 감소해주는 식으로하여 q가 비었을 때 cnt 값에 따라 결과를 반환해줄 수 있을 것 같습니다!
BFS 문제들은 조금만 난이도가 올라가게 되면 예외 처리를 신중하게 잘 찾아야하는데 깔끔하게 잘 찾아내서 보기 좋았던 것 같습니다!
토마토는 3차원도 존재한다는 거 아시나요? 맛있는 토마토! 문제 추천 드립니다!

[seongwon030]

익은 토마토의 위치만 고려해서 시간복잡도를 고려하지는 못했네요.. 감사합니다!! 3차원도 한 번 풀러 가보겠습니다 ㅎㅎ

[yuyu0830]

11차원 토마토도 있다는 것 아시나요? 하이퍼 토마토
3차원 토마토 원리를 이해하시면 금방 푸실 수 있으실 거예요 ㅎㅎ

다른 코드 부분도 이해하기 쉬워서 좋았습니다.
아실 수도 있지만 파이썬은 for-else 라는 재밌는 문법이 있더라구요.
잘 사용하면 flag 변수 없이도 짤 수 있을 것 같아요!

[yuyu0830]

코드 잘 봤습니다 :)

[seongwon030]

11차원 토마토도 있다는 것 아시나요? 하이퍼 토마토 3차원 토마토 원리를 이해하시면 금방 푸실 수 있으실 거예요 ㅎㅎ

다른 코드 부분도 이해하기 쉬워서 좋았습니다. 아실 수도 있지만 파이썬은 for-else 라는 재밌는 문법이 있더라구요. 잘 사용하면 flag 변수 없이도 짤 수 있을 것 같아요!

다음에 flag없이도 짜 봐야겠네요! 문제 추천 감사합니다 ㅎㅎ

[dhlee777]

저도 얼마전에 풀었던 문제인데 bfs를 해주고나서 안익은 토마토가 있는지 확인하기 위해
다시 탐색하는 과정에서 좀 힘들었던 경험이있네요.. 깔끔한 코드 잘 보고 갑니다 !