졍's 기술 블로그

음식물 피하기

티어 : Silver 1
시간 제한 : 2 초
메모리 제한 : 128 MB
알고리즘 분류 : 그래프 이론, 그래프 탐색, 너비 우선 탐색, 깊이 우선 탐색

문제

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

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

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

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

입력

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

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

출력

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

예제 입출력

Algorithm

DFS - 재귀
1. 쓰레기가 떨어져있는 곳은 1, 쓰레기가 없는 곳은 0으로 두고 graph 구현
2. (0, 0) 부터 (N-1, M-1) 까지 DFS를 이용해 덩어리 당 음식물 개수 확인
3. DFS에서 True 반환될 때마다 음식물 개수 최댓값으로 갱신

Code

def dfs(x, y):
    global count_
    
    # graph의 범위를 벗어나면 False Return
    if x < 0 or x > N-1 or y < 0 or y > M-1:
        return False
    
    # 현재 위치에 방문한 적이 있거나 음식물이 없는 곳이면 False Return
    if graph[x][y] == 0:
        return False
    
    # 현재 위치 방문 기록
    graph[x][y] = 0
    count_ += 1
    
    # 인접 노드 방문
    dfs(x-1, y)
    dfs(x+1, y)
    dfs(x, y-1)
    dfs(x, y+1)
    
    return True
    
import sys
input = sys.stdin.readline

# 입력
N, M, K = map(int, input().split())
sys.setrecursionlimit(N*M*K)
graph = [[0 for _ in range(M)] for _ in range(N)]
for _ in range(K):
    x, y = map(int, input().split())
    graph[x-1][y-1] = 1

global count_ # 덩어리에 속해있는 노드 개수
answer = 0
for x in range(N):
    for y in range(M):
        count_ = 0
        # DFS 반환값이 True일 때만 정답 갱신
        if dfs(x, y):
            answer = max(answer, count_)
print(answer)

메모리: 35760 KB
시간: 80 ms