[Python] 백준 1734 음식물 피하기 - BFS

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

1743번: 음식물 피하기

 

문제

코레스코 콘도미니엄 8층은 학생들이 3끼의 식사를 해결하는 공간이다. 그러나 몇몇 비양심적인 학생들의 만행으로 음식물이 통로 중간 중간에 떨어져 있다. 이러한 음식물들은 근처에 있는 것끼리 뭉치게 돼서 큰 음식물 쓰레기가 된다. 

이 문제를 출제한 선생님은 개인적으로 이러한 음식물을 실내화에 묻히는 것을 정말 진정으로 싫어한다. 참고로 우리가 구해야 할 답은 이 문제를 낸 조교를 맞추는 것이 아니다. 

통로에 떨어진 음식물을 피해가기란 쉬운 일이 아니다. 따라서 선생님은 떨어진 음식물 중에 제일 큰 음식물만은 피해 가려고 한다. 

선생님을 도와 제일 큰 음식물의 크기를 구해서 “10ra"를 외치지 않게 도와주자.

입력

첫째 줄에 통로의 세로 길이 N(1 ≤ N ≤ 100)과 가로 길이 M(1 ≤ M ≤ 100) 그리고 음식물 쓰레기의 개수 K(1 ≤ K ≤ N×M)이 주어진다.  그리고 다음 K개의 줄에 음식물이 떨어진 좌표 (r, c)가 주어진다.

좌표 (r, c)의 r은 위에서부터, c는 왼쪽에서부터가 기준이다. 입력으로 주어지는 좌표는 중복되지 않는다.

출력

첫째 줄에 음식물 중 가장 큰 음식물의 크기를 출력하라.

 

예제 입력 1 복사

3 4 5
3 2
2 2
3 1
2 3
1 1

예제 출력 1 복사

4

힌트

# . . .
. # # .
# # . .

위와 같이 음식물이 떨어져있고 제일큰 음식물의 크기는 4가 된다. (인접한 것은 붙어서 크게 된다고 나와 있음. 대각선으로는 음식물 끼리 붙을수 없고 상하좌우로만 붙을수 있다.)

---

풀이

•  BFS 사용하여 음식물의 개수 배열에 저장
• 대각선으로는 음식물 덩어리가 될 수 없으므로 상하좌우 방향벡터 정의하여 탐색
• 음식물이 존재할 때, 영역에 존재하는 음식물 개수를 증가하여 반환
• 영역에 존재하는 총 음식물 개수(크기)를 배열에 담고 max 함수 사용하여 가장 큰 음식물 크기 출력

코드

from collections import deque;

#  상하좌우 방향벡터
dx = [0, 0, 1, -1]
dy = [1, -1, 0, 0]

# 음식물 담을 리스트
result = []

def Bfs(x, y):
    cnt = 1
    q = deque()
    q.append((x, y))
    visited[x][y] = True

    while q:
        x, y = q.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]

            # 탐색 범위 제한
            if nx < 0 or nx >= N or ny < 0 or ny >= M:
                continue
            # 음식물 쓰레기가 있고 방문하지 않았다면 개수 증가
            if board[nx][ny] == '#' and not visited[nx][ny]:
                q.append((nx, ny))
                visited[nx][ny] = True
                cnt += 1
    return cnt

if __name__ == "__main__":
    # N: 세로 M: 가로 K: 음식물 개수
    N, M, K = map(int, input().split())
    board = [['.'] * M for _ in range(N)]
    visited = [[False] * M for _ in range(N)]

    # 음식물 위치 받아오기
    # r, c는 좌표이므로 배열의 인덱스를 받아오기 위해 -1 해준다
    for _ in range(K):
        r, c = map(int, input().split())
        board[r-1][c-1] = '#'

    for i in range(N):
        for j in range(M):
            # 음식물 있으면 배열에 음식물 개수 저장
            if board[i][j] == '#' and not visited[i][j]:
                result.append(Bfs(i, j))

    # 배열에서 가장 큰 음식물 출력
    print(max(result))