Algorithm PS/DFS & BFS

[BOJ/백준] 4963번 섬의 개수 (Python 파이썬)

great_park 2023. 1. 27. 17:12
반응형

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

 

4963번: 섬의 개수

입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스의 첫째 줄에는 지도의 너비 w와 높이 h가 주어진다. w와 h는 50보다 작거나 같은 양의 정수이다. 둘째 줄부터 h개 줄에는 지도

www.acmicpc.net

 

문제

정사각형으로 이루어져 있는 섬과 바다 지도가 주어진다. 섬의 개수를 세는 프로그램을 작성하시오.

한 정사각형과 가로, 세로 또는 대각선으로 연결되어 있는 사각형은 걸어갈 수 있는 사각형이다. 

두 정사각형이 같은 섬에 있으려면, 한 정사각형에서 다른 정사각형으로 걸어서 갈 수 있는 경로가 있어야 한다. 지도는 바다로 둘러싸여 있으며, 지도 밖으로 나갈 수 없다.

입력

입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스의 첫째 줄에는 지도의 너비 w와 높이 h가 주어진다. w와 h는 50보다 작거나 같은 양의 정수이다.

둘째 줄부터 h개 줄에는 지도가 주어진다. 1은 땅, 0은 바다이다.

입력의 마지막 줄에는 0이 두 개 주어진다.

출력

각 테스트 케이스에 대해서, 섬의 개수를 출력한다.

예제 입력 1

1 1
0
2 2
0 1
1 0
3 2
1 1 1
1 1 1
5 4
1 0 1 0 0
1 0 0 0 0
1 0 1 0 1
1 0 0 1 0
5 4
1 1 1 0 1
1 0 1 0 1
1 0 1 0 1
1 0 1 1 1
5 5
1 0 1 0 1
0 0 0 0 0
1 0 1 0 1
0 0 0 0 0
1 0 1 0 1
0 0

예제 출력 1

0
1
1
3
1
9

 


풀이

문제를 살펴보면, 어떤 Land영역 (1)에 대해서 해당 지점에서 갈 수 있는 모든 영역을 BFS로 모두 탐색하고, 이 과정을 완료하면 섬 하나를 count한 것과 같게 됩니다.

따라서 주어진 모든 영역에 대해서 이중 loop을 돌며  BFS를 시작할 지점인 Land 영역을 하나 찾고, BFS를 실시한다면 모든 섬의 개수를 찾을 수 있을 것입니다. 

다른 문제들과 조금은 차이점이 있다면 아래 2가지 입니다.
1. 대각선까지 고려한 8방향이라는 점
2. test case 개수가 주어지지 않고 (0, 0)이 입력되면 종료하는 점

(1)의 경우 대각선 4방향에 대해서 변화량을 추가하면 되고, (2)의 경우 while문에서 (0,0)이 입력으로 들어오면 break을 걸어주면 됩니다.

BFS을 구현하면 아래와 같습니다.

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

# 시작점이 주어지면 이동가능한 모든 점을 "방문했음" 상태로 만들기 -> 0
def BFS(x, y):
  queue = deque()
  queue.append([x,y])
  graph[y][x] = 0 # 방문 처리
  
  while queue:
    x, y = queue.popleft()
    for i in range(8):
      next_x = x + dx[i]
      next_y = y + dy[i]
      
      if 0 <= next_x < w and 0 <= next_y < h: # 범위 확인
        if graph[next_y][next_x] == 1: # 이동 가능 여부 확인
          queue.append([next_x, next_y])
          graph[next_y][next_x] = 0

첫 시작점을 기준으로 8방향에 대해서 범위와 이동 가능 여부를 확인한 후 큐에 넣고 방문 처리를 하는 식으로 진행됩니다.

이를 바탕으로 최종적인 구현은 아래와 같습니다.

Code

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

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

# 시작점이 주어지면 이동가능한 모든 점을 "방문했음" 상태로 만들기 -> 0
def BFS(x, y):
  queue = deque()
  queue.append([x,y])
  graph[y][x] = 0 # 방문 처리
  
  while queue:
    x, y = queue.popleft()
    for i in range(8):
      next_x = x + dx[i]
      next_y = y + dy[i]
      
      if 0 <= next_x < w and 0 <= next_y < h: # 범위 확인
        if graph[next_y][next_x] == 1: # 이동 가능 여부 확인
          queue.append([next_x, next_y])
          graph[next_y][next_x] = 0


# main
while(True):
  graph = list()
  w, h = map(int, input().split())
  
  # 0 0이 입력되면 종료된다.
  if (w == 0 and h == 0):
    break
  
  cnt = 0 # 섬 개수
  # input graph를 초기화 한다.
  for i in range(h):
    graph.append(list(map(int, input().split())))

  for i in range(h): # y
    for j in range(w): # x
      if graph[i][j] == 1:
        BFS(j, i)
        cnt += 1
  print(cnt)
반응형