[baekjoon]python #2667 단지번호붙이기

[baekjoon]python #2667 단지번호붙이기

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

2667번: 단지번호붙이기

import sys
input = sys.stdin.readline
from collections import deque

n = int(input())
arr = [list(map(int,input().strip())) for _ in range(n)]

#집의 수
answer = []
#단지 수
cnt=0


def bfs(y,x):
    queue = deque()
    queue.append((y,x))
    # 이미 다녀간 곳 체크
    arr[y][x] = 0
	
    # 집의 수 체크
    bfs_cnt = 0
    while queue:
        ny,nx = queue.popleft()

        directy = [-1,1,0,0]
        directx = [0,0,-1,1]

        for i in range(4):
            dy = directy[i] + ny
            dx = directx[i] + nx

            if 0<=dy<n and 0<=dx<n:
                if arr[dy][dx]==1:
                	#다녀간 곳 체크
                    arr[dy][dx] = 0
                    queue.append((dy,dx))
        bfs_cnt+=1
    answer.append(bfs_cnt)

for i in range(n):
    for j in range(n):
        if arr[i][j] == 1:
            bfs(i,j)
            cnt+=1

#오름차순
answer.sort()

#출력
print(cnt)
for a in answer:
    print(a)