[baekjoon] 백준 1926번(파이썬): 그림

문제

1926번: 그림

- 어떤 큰 도화지에 그림이 그려져 있을 때, 그 그림의 개수와, 그 그림 중 넓이가 가장 넓은 것의 넓이를 출력하여라.

- 단, 그림이라는 것은 1로 연결된 것을 한 그림이라고 정의하자.

- 가로나 세로로 연결된 것은 연결이 된 것이고 대각선으로 연결이 된 것은 떨어진 그림이다.

- 그림의 넓이란 그림에 포함된 1의 개수이다.

- 도화지의 세로 크기 n(1 ≤ n ≤ 500)과 가로 크기 m(1 ≤ m ≤ 500)이 차례로 주어진다.

- 그림의 정보가 주어진다. (단 그림의 정보는 0과 1이 공백을 두고 주어지며, 0은 색칠이 안된 부분, 1은 색칠이 된 부분을 의미한다)

- 그림의 개수, 둘째 줄에는 그 중 가장 넓은 그림의 넓이를 출력하여라.

- 단, 그림이 하나도 없는 경우에는 가장 넓은 그림의 넓이는 0이다.

알고리즘

- dfs 탐색을 통해 문제를 수행한다.

- 문제 조건 중에 그림이 하나도 없는 경우에는 가장 넓은 그림의 넓이는 0이라고 했기 때문에 조건문을 통해 그림의 개수가 0일 때 가장 넓은 그림의 넓이를 0이 출력되게 한다.

코드

import sys
sys.setrecursionlimit(10 ** 6)


# df 탐색
def dfs(x, y):

    if x <= -1 or x >= n or y <= -1 or y >= m:
        return False

    if graph[x][y] == 1:

        # 그림 개수 카운트
        cnt.append(1)

        # 해당 노드 방문 처리
        graph[x][y] = 0

        # 상/하/좌/우 재귀적으로 탐색
        dfs(x + 1, y)
        dfs(x - 1, y)
        dfs(x, y + 1)
        dfs(x, y - 1)

        return True
    return False


n, m = map(int, sys.stdin.readline().split())

# 각 노드의 연결된 정보를 표현
graph = [list(map(int, sys.stdin.readline().split())) for _ in range(n)]

cnt = [] # 그림의 개수 카운트
res = [] # 각 구역의 연결된 그림의 크기

for i in range(n):
    for j in range(m):
        if graph[i][j] == 1:
            dfs(i, j)
            res.append(len(cnt))
            cnt = []

# 그림의 개수 출력
print(len(res))

# 그림의 개수가 있으면 제일 큰 값을 출력
# 그게 아니라면 0을 출력
if len(res):
    print(max(res))
else:
    print(0)

github

GitHub - junjange/CodingTest: 내가 푼 코딩 테스트 문제와 해결법