[BOJ] 2667번: 단지번호붙이기

✏️ 문제

<그림 1>과 같이 정사각형 모양의 지도가 있다. 1은 집이 있는 곳을, 0은 집이 없는 곳을 나타낸다. 철수는 이 지도를 가지고 연결된 집의 모임인 단지를 정의하고, 단지에 번호를 붙이려 한다. 여기서 연결되었다는 것은 어떤 집이 좌우, 혹은 아래위로 다른 집이 있는 경우를 말한다. 대각선상에 집이 있는 경우는 연결된 것이 아니다. <그림 2>는 <그림 1>을 단지별로 번호를 붙인 것이다. 지도를 입력하여 단지수를 출력하고, 각 단지에 속하는 집의 수를 오름차순으로 정렬하여 출력하는 프로그램을 작성하시오.

 

 

🤖 입출력 예시

 

💡 풀이

BFS+DFS 문제집 가지고 그래프 유형 열심히 풀어보는 중이다. 몇 번의 코테를 경험해본 결과, 나는 기본 유형에 조건이 하나만 추가돼도 손을 못 댄다는 걸 알았기 때문에 이 문제처럼 추가적인 처리(단지 내 집의 수를 세는 것)를 해줘야 하는 유형을 많이 풀어봐야겠다.

1) DFS 풀이

import sys

sys.setrecursionlimit(10000)
    
dx = [0, 0, -1, 1]
dy = [-1, 1, 0, 0]
            
N = int(input())

board = [list(map(int, input())) for _ in range(N)]

visited = [[False]*N for _ in range(N)]

town = []

def DFS(row, col):
    global count
    
    count += 1

    visited[row][col] = True
    
    for i in range(4):
        new_row, new_col = row+dx[i], col+dy[i]
        
        if (0 <= new_row < len(board) and 0 <= new_col < len(board)) and visited[new_row][new_col] == 0 and board[new_row][new_col] == 1:
            DFS(new_row, new_col)

    return count

for row in range(N):
    for col in range(N):
        if board[row][col] == 1 and visited[row][col] == 0:
            count = 0
            
            town.append(DFS(row, col))

print(len(town))

for home in sorted(town):
    print(home)

 

2) BFS 풀이

from collections import deque

dx = [0, 0, -1, 1]
dy = [-1, 1, 0, 0]
            
N = int(input())

board = [list(map(int, input())) for _ in range(N)]

visited = [[False]*N for _ in range(N)]

town = []

def BFS(row, col):
    global count

    queue = deque([(row, col)])
    
    visited[row][col] = True
    
    count += 1  # 자기 자신 체크
    
    while queue:
        curr_row, curr_col = queue.popleft()
        
        for i in range(4):
            new_row, new_col = curr_row+dx[i], curr_col+dy[i]
            
            if (0 <= new_row < len(board) and 0 <= new_col < len(board)) and visited[new_row][new_col] == 0 and board[new_row][new_col] == 1:
                count += 1
                
                visited[new_row][new_col] = True
                
                queue.append((new_row, new_col))
                
    return count

for row in range(N):
    for col in range(N):
        if board[row][col] == 1 and visited[row][col] == 0:
            count = 0
            
            town.append(BFS(row, col))

print(len(town))

for home in sorted(town):
    print(home)

 

BFS로 풀 때 자기 자신을 체크하는 걸 빼먹어서 단지 내 집의 수가 정답보다 1씩 작게 나오는 문제가 있었지만 해결!

 

 

32ms가 DFS 풀이, 52ms가 BFS 풀이다. 근데 DFS/BFS 공부할 때 BFS가 더 빠르다고 했는데 처리 범위가 커지면 BFS의 속도 체감이 될수도 있지만 아직까지는 DFS를 사용한 풀이가 더 빠른 경우가 많았다.