[백준/BOJ] silver1 - 1743번 음식물 피하기 (Python)

▶1743 - 음식물 피하기

백준 로고

▶문제

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

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

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

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

 

▶입력

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

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

 

▶출력

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

728x90

▶풀이

bfs 중에서도 쉬운 문제인 것 같다.

기본적인 bfs를 그대로 사용하면 되고, 해당 음식물이 쌓이는 동안 cnt를 계산해 return 해주면 된다.

 

전체 arr에서도 음식물을 만날 때마다 해주고, return 받은 값 중에서 최댓값을 출력해 주면 된다.

visit를 이용해 이전에 다른 곳에서 출발해 만난 곳은 계산 안 하도록 설정하면 된다.

from collections import deque


def bfs(a, b):
    queue = deque()
    queue.append((a, b))
    visit[a][b] = 1
    cnt = 1

    while queue:
        x, y = queue.popleft()

        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]

            if 0 <= nx < n and 0 <= ny < m:
                if arr[nx][ny] == 1 and visit[nx][ny] == 0:
                    visit[nx][ny] = 1
                    queue.append((nx, ny))
                    cnt += 1

    return cnt


n, m, k = map(int, input().split())

arr = [[0 for _ in range(m)] for _ in range(n)]
visit = [[0 for _ in range(m)] for _ in range(n)]
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
answer = 0

for i in range(k):
    r, c = map(int, input().split())
    arr[r - 1][c - 1] = 1

for i in range(n):
    for j in range(m):
        if arr[i][j] == 1 and visit[i][j] == 0:
            tmp = bfs(i, j)
            answer = max(answer, tmp)

print(answer)