알고리즘 연습/DFS와 BFS

[🥈1 / 백준 1926 / 파이썬] 그림

김세진 2022. 2. 7. 09:40
반응형

 

 

1926번: 그림

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

www.acmicpc.net

 

문제

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

입력

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

출력

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

 

예제 입력

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

예제 출력

4
9





 

풀이

 

DFS, 혹은 BFS로 영역의 개수와 최대 넓이를 출력하는 문제이다.

필자는 BFS로 해결했으므로 BFS 코드를 첨부한다.

 

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

def main():
    n,m = map(int,input().split())
    board = [list(map(int,input().split())) for _ in range(n)]
    visited = [[False]*m for _ in range(n)]
    move = [(0,1),(1,0),(0,-1),(-1,0)]
    q = deque()

    num_area,max_area = 0,0
    for i in range(n):
        for j in range(m):
            area = 0
            if visited[i][j]:
                continue
            if board[i][j]:
                q.append((i,j))
                visited[i][j] = True
                num_area += 1
                area += 1

            while q:
                x,y = q.popleft()
                for dx,dy in move:
                    nx = x+dx; ny = y+dy
                    if n>nx>=0 and m>ny>=0 and board[nx][ny] and not visited[nx][ny]:
                        visited[nx][ny] = True
                        area += 1
                        q.append((nx,ny))
            max_area = max(area,max_area)
    print(num_area,max_area,sep="\n")
if __name__ == "__main__":
    main()
반응형