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

📖 문제

📃 코드

import sys
input = sys.stdin.readline

N = int(input())

complex = []
result = []
cnt = 0

# 아래, 위, 왼쪽, 오른쪽
dx = [0, 0, -1, 1]
dy = [1, -1, 0, 0]

def bfs(x, y):
    global cnt
    if x < 0 or x >= N or y < 0 or y >= N:
        return
    if complex[x][y] == 1:
        cnt += 1
        # 다시 방문하지 않게 초기화
        complex[x][y] = 0
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            bfs(nx, ny)


for i in range(N):
    complex.append(list(map(int, input().rstrip())))

for row in range(N):
    for col in range(N):
        if complex[row][col] == 1:
            bfs(row, col)
            result.append(cnt)
            cnt = 0

result.sort()
print(len(result))
print("\n".join(map(str,result)))

 

🔗 링크

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

2667번: 단지번호붙이기