[백준 1743번] 파이썬 - 음식물 피하기

백준 1743 - 음식물 피하기

시간 제한 2초, 메모리 제한 128MB

# 조건

• 코레스코 콘도미니엄 8층은 학생들이 3끼의 식사를 해결하는 공간이다.
• 그러나 몇몇 비양심적인 학생들의 만행으로 음식물이 통로 중간 중간에 떨어져 있다.
• 이러한 음식물들은 근처에 있는 것끼리 뭉치게 돼서 큰 음식물 쓰레기가 된다.
• 이 문제를 출제한 선생님은 개인적으로 이러한 음식물을 실내화에 묻히는 것을 정말 진정으로 싫어한다.
  • 참고로 우리가 구해야 할 답은 이 문제를 낸 조교를 맞추는 것이 아니다.
• 통로에 떨어진 음식물을 피해가기란 쉬운 일이 아니다.
• 따라서 선생님은 떨어진 음식물 중에 제일 큰 음식물만은 피해 가려고 한다.
• 선생님을 도와 제일 큰 음식물의 크기를 구해서 “10ra"를 외치지 않게 도와주자.

입력

• 첫째 줄에 통로의 세로 길이 N(1 ≤ N ≤ 100)과 가로 길이 M(1 ≤ M ≤ 100) 그리고 음식물 쓰레기의 개수 K(1 ≤ K ≤ N×M)이 주어진다.
• 그리고 다음 K개의 줄에 음식물이 떨어진 좌표 (r, c)가 주어진다.
• 좌표 (r, c)의 r은 위에서부터, c는 왼쪽에서부터가 기준이다.
• 입력으로 주어지는 좌표는 중복되지 않는다.

출력

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

# 접근 방법

• bfs를 이용해서 풀어주면 된다.
• 각 음식물이 있는 위치를 arr에 표시해주고, 방문하지 않은 음식물이 있다면 bfs 함수를 실행해준다.
• 주변의 음식물 개수를 count 해준 후 그 크기를 return하여 result를 max로 갱신해주면 된다.

import sys
sys.stdin = open('input.txt')
input = sys.stdin.readline
from collections import deque

def bfs(bi, bj):
    q = deque()
    q.append((bi, bj))
    cnt = 1
    while q:
        si, sj = q.popleft()
        for d in range(4):
            ni, nj = si+di[d], sj+dj[d]
            if 0<=ni<N and 0<=nj<M and visited[ni][nj] == 0 and arr[ni][nj] == 1:
                q.append((ni, nj))
                visited[ni][nj] = 1
                cnt += 1
    return cnt

di, dj = [1, -1, 0, 0], [0, 0, 1, -1]
N, M, K = map(int, input().split())
arr = [[0] * M for _ in range(N)]
visited = [[0] * M for _ in range(N)]
for _ in range(K):
    r, c = map(int, input().split())
    arr[r-1][c-1] = 1

result = 0
for i in range(N):
    for j in range(M):
        if arr[i][j] == 1 and visited[i][j] == 0:
            visited[i][j] = 1
            cnt = bfs(i, j)
            result = max(result, cnt)
print(result)