백준 2667번 : 단지번호붙이기(python)

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

2667번: 단지번호붙이기

 

n = int(input())
data = [list(map(int, input())) for _ in range(n)]
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]


def dfs(x, y):
    count = 1
    data[x][y] = 0
    q = [(x, y)]
    while q:
        x, y = q.pop()
        for i in range(4):
            nx, ny = x + dx[i], y + dy[i]
            if 0 <= nx < n and 0 <= ny < n and data[nx][ny] == 1:
                q.append((nx,ny))
                data[nx][ny] = 0
                count += 1
    return count


result = 0
number = []
for x in range(n):
    for y in range(n):
        if data[x][y] == 1:
            result += 1
            c = dfs(x, y)
            number.append(c)

number.sort()
print(result)
for num in number:
     print(num)