[Python] 백준 1926번 - 그림 (실버 1)

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

1926번: 그림

 

혼자 힘으로 풀었는가? O

알고리즘 분류
 - 그래프 이론
 - 그래프 탐색
 - 너비 우선 탐색 (BFS)
 - 깊이 우선 탐색 (DFS)

 

 

문제

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

입력

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

출력

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

 

---

 

BFS

import sys
input = sys.stdin.readline

n, m = map(int, input().split())
data = []
for i in range(n):
    data.append(list(map(int, input().split())))
    
visited = [ [0 for _ in range(m)] for _ in range(n)]

result = []

dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]

from collections import deque

def bfs(x, y):
    q = deque()
    q.append((x, y))
    visited[x][y] = 1
    cnt = 1
    while q:
        x, y = q.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < n and 0 <= ny < m:
                if data[nx][ny] == 1 and visited[nx][ny] == 0:
                    visited[nx][ny] = 1
                    cnt += 1
                    q.append((nx, ny))
                    
    return cnt
for i in range(n):
    for j in range(m):
        if data[i][j] == 1 and visited[i][j] == 0:
            cnt = bfs(i, j)
            result.append(cnt)
            
print(len(result))
if len(result) == 0:
    print(0)
else:
    print(max(result))

 

 

DFS

 

import sys
input = sys.stdin.readline

sys.setrecursionlimit(10**9)

n, m = map(int, input().split())
data = []
for i in range(n):
    data.append(list(map(int, input().split())))
    
visited = [ [0 for _ in range(m)] for _ in range(n)]

result = []

dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]

def dfs(x, y):
    global cnt
    cnt += 1
    visited[x][y] = 1
    for i in range(4):
        nx = x + dx[i]
        ny = y + dy[i]
        if 0 <= nx < n and 0 <= ny < m:
            if data[nx][ny] == 1 and visited[nx][ny] == 0:
                dfs(nx, ny)
                
                
for i in range(n):
    for j in range(m):
        if data[i][j] == 1 and visited[i][j] == 0:
            cnt = 0
            dfs(i, j)
            result.append(cnt)
print(len(result))
if len(result) == 0:
    print(0)
else:
    print(max(result))