본문 바로가기
백준알고리즘/DFS 와 BFS

(Python/🥈3)백준 알고리즘 1926번: 그림

by windy7271 2023. 7. 6.
728x90
반응형

 

문제 바로가기 

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

문제:

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

입력:

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

 

출력:

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

 

 

풀이 :

 

import sys
from collections import deque

sys.stdin = open('/Users/song/Desktop/Python/Python/h.txt', 'r')

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

# 그래프를 표현
graph = [list(map(int, sys.stdin.readline().rstrip().split())) for _ in range(n)]

# 이동 방향
move = [(1, 0), (-1, 0), (0, -1), (0, 1)]

# 방문 여부
visited = [[False] * m for _ in range(n)]


# BFS 함수를 정의합니다.
def bfs(x, y):
    q = deque([(x, y)])  # 시작 위치
    visited[x][y] = True  # 시작 위치를 방문
    cnt = 1  # 방문한 노드의 개수

    # 큐가 빌 때까지 반복합니다.
    while q:
        a, b = q.popleft()  

        # 이동할 수 있는 모든 방향에 대해 확인합니다.
        for dx, dy in move:
            nx = dx + a
            ny = dy + b

            # 다음 위치가 그래프 내에 있고, 연결된 구성 요소에 속하며 아직 방문하지 않았다면,
            if 0 <= nx < n and 0 <= ny < m and graph[nx][ny] == 1 and not visited[nx][ny]:
                visited[nx][ny] = True  # 다음 위치를 방문했다고 표시.
                q.append((nx, ny))  # 다음 위치를 큐에 추가.
                cnt += 1  # 방문한 노드의 개수를 1 증가.

    return cnt  # BFS를 마친 후 방문한 노드의 개수를 반환

cnt = 0  # 연결된 구성 요소의 개수를 세기 위한 변수입니다.
lst = 0  # 가장 큰 구성 요소의 크기를 저장하기 위한 변수입니다.

# 모든 위치를 순회하면서 연결된 구성 요소를 탐색합니다.
for i in range(n):
    for j in range(m):
        # 현재 위치가 연결된 구성 요소에 속하고 아직 방문하지 않았다면,
        if graph[i][j] == 1 and not visited[i][j]:
            lst = max(lst, bfs(i, j))  # BFS를 수행하고 가장 큰 구성 요소의 크기 업데이트
            cnt += 1  # 연결된 구성 요소의 개수를 1 증가

print(cnt)  # 연결된 구성 요소
print(lst)  # 최대 그림갯수

 

기본적인 bfs형태가 아닌가 싶다 

방문처리를 해줘야지 이미 만든 그림은 방문하지 않는다.

 

 

반응형

댓글