[BOJ/백준] 1926번 그림 (Python 파이썬)

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

1926번: 그림

 

문제

어떤 큰 도화지에 그림이 그려져 있을 때, 그 그림의 개수와, 그 그림 중 넓이가 가장 넓은 것의 넓이를 출력하여라. 단, 그림이라는 것은 1로 연결된 것을 한 그림이라고 정의하자. 가로나 세로로 연결된 것은 연결이 된 것이고 대각선으로 연결이 된 것은 떨어진 그림이다. 그림의 넓이란 그림에 포함된 1의 개수이다.

입력

첫째 줄에 도화지의 세로 크기 n(1 ≤ n ≤ 500)과 가로 크기 m(1 ≤ m ≤ 500)이 차례로 주어진다. 두 번째 줄부터 n+1 줄 까지 그림의 정보가 주어진다. (단 그림의 정보는 0과 1이 공백을 두고 주어지며, 0은 색칠이 안된 부분, 1은 색칠이 된 부분을 의미한다)

출력

첫째 줄에는 그림의 개수, 둘째 줄에는 그 중 가장 넓은 그림의 넓이를 출력하여라. 단, 그림이 하나도 없는 경우에는 가장 넓은 그림의 넓이는 0이다.

예제 입력 1

6 5
1 1 0 1 1
0 1 1 0 0
0 0 0 0 0
1 0 1 1 1
0 0 1 1 1
0 0 1 1 1

예제 출력 1

4
9

 

---

풀이

문제를 살펴보면, 1로 연결된 각 그림의 개수와, 가장 넓은 그림의 넓이를 구해야 합니다. 따라서, 주어진 그래프에 대해서, (1) 아직 방문하지 않았고 (2) 방문 가능한 구역 [= 1] 일 때 탐색을 시작해서, 갈 수 있는 곳까지 모두 방문한 뒤 BFS의 depth을 활용하여 탐색한 그림의 넓이를 구할 것입니다.

이때, 그림이 하나도 없는 경우, 개수는 0, 넓이도 0을 출력해야 합니다. 따라서 이에 대한 처리를 해줘야 합니다.

또한, 아래의 제 풀이에서 따로 방문 여부 리스트를 만들지 않고, 그래프에 그대로 탐색의 depth을 저장함으로써 방문 여부를 체크(graph[y][x] == 1)함과 동시에 최종적으로 탐색의 넓이를 구할 수 있습니다. 이때, 하필이면 그림의 구역이 1로 주어졌기 때문에 가장 넓은 넓이가 1인 경우를 제외하고는 모두 1을 빼줘야 test case을 통과할 수 있습니다. 해당 부분은 직접 디버깅해보시면서 구현해 보시면 이해가 되실 것 같습니다. 

 

Code

import sys
from collections import deque
input = sys.stdin.readline
N, M = map(int, input().split())
graph = [list(map(int, input().split())) for _ in range(N)]

dx = [0, 0, 1, -1]
dy = [1, -1, 0, 0]
answer = list()


def BFS(x, y):
    queue = deque([(x, y)])
    graph[y][x] = 1
    size = 1

    while queue:
        x, y = queue.popleft()
        for i in range(4):
            next_x, next_y = x + dx[i], y + dy[i]
            
            if 0 <= next_x < M and 0 <= next_y < N: #범위 확인
                if graph[next_y][next_x] == 1: # 진입 가능 확인
                    queue.append([next_x, next_y])
                    graph[next_y][next_x] = 0
                    size += 1
    return size

for i in range(N):
    for j in range(M):
        if graph[i][j] == 1: # 그림 구역이면 탐색 진입
            answer.append(BFS(j, i))
            
print(len(answer))
if len(answer) == 0:
    print(0)
else:
    if max(answer) == 1:
        print(max(answer))
    else:
        print(max(answer) - 1)