[백준] 2667 - 단지번호붙이기 (파이썬, Python)

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

2667번: 단지번호붙이기

from collections import deque
import sys

def solution(i, j, cnt) :
    cnt += 1
    matrix[i][j] = 0

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

    queue = deque([(i, j)])
    
    while queue:
        pos = queue.popleft()
        for k in range(4):
            cx = pos[0] + dx[k]
            cy = pos[1] + dy[k]
            if(0 <= cx and cx < N and 0 <= cy and cy < N):
                if(matrix[cx][cy] != 0):
                    matrix[cx][cy] = 0
                    cnt += 1
                    queue.append((cx, cy))
    return cnt
N = int(sys.stdin.readline().rstrip())

matrix = [[int(i) for i in sys.stdin.readline().rstrip()]for _ in range(N)]
dong = []

for i in range(N):
    for j in range(N):
        cnt = 0
        if(matrix[i][j] != 0):
            dong.append(solution(i, j, cnt))

print(len(dong))

dong.sort()

for i in dong:
    print(i)