[ 2023-07-22 🦖 TIL ]

백준 2667번 파이썬

문제

코드

import sys
from collections import deque

input = sys.stdin.readline

n = int(input())
feild = []
visited = [[0] * n for _ in range(n)]

for _ in range(n):
  feild.append(list(map(int, input().strip())))

result = []
move = ((-1,0),(1,0),(0,-1), (0,1))
# 인덱스 기준 (상, 하, 좌, 우)

def bfs(si, sj):
  q = deque([(si, sj)])
  visited[si][sj] = 1
  cnt = 1

  while q : 
    ci, cj = q.popleft()
    for di, dj in move: # 상하좌우 살피기에 4번 반복
      ni, nj = ci+di, cj+dj 
      # 현재 위치에서 상하좌우 좌표를 확인한다.
      if ((0<=ni<n) and (0<=nj<n) and 
      		visited[ni][nj] == 0 and feild[ni][nj] == 1):
        # ni,nj 값이 너비(n)을 벗어나지 않고 상하좌우 중에 방문하지
        # 않은 곳이 있으면 그 해당 좌표로 이동하여 살핀다.
        q.append((ni, nj))
        visited[ni][nj] = 1
        cnt += 1
  # 상하좌우에 이제 0밖에 없으면 반복문 끝
  
  return cnt

for i in range(n):
  for j in range(n):
    if feild[i][j] == 1 and visited[i][j] == 0 :
      result.append(bfs(i, j))
      
print(len(result))
result.sort()
for dangi in result:
  print(dangi)

처음에 move = ((-1,0),(1,0),(0,-1), (0,1)) 이 부분이 이해되지 않아서 고생했지만 계속 보고 코드를 직접 타이핑 해보면서 확인해보니 이해 할 수 있었습니다.
이 풀이를 처음 생각하신 분 천재인듯...🥹