[백준] 1937번 - 욕심쟁이 판다

📩 출처

문제

n × n의 크기의 대나무 숲이 있다. 욕심쟁이 판다는 어떤 지역에서 대나무를 먹기 시작한다. 그리고 그 곳의 대나무를 다 먹어 치우면 상, 하, 좌, 우 중 한 곳으로 이동을 한다. 그리고 또 그곳에서 대나무를 먹는다. 그런데 단 조건이 있다. 이 판다는 매우 욕심이 많아서 대나무를 먹고 자리를 옮기면 그 옮긴 지역에 그 전 지역보다 대나무가 많이 있어야 한다.

이 판다의 사육사는 이런 판다를 대나무 숲에 풀어 놓아야 하는데, 어떤 지점에 처음에 풀어 놓아야 하고, 어떤 곳으로 이동을 시켜야 판다가 최대한 많은 칸을 방문할 수 있는지 고민에 빠져 있다. 우리의 임무는 이 사육사를 도와주는 것이다. n × n 크기의 대나무 숲이 주어져 있을 때, 이 판다가 최대한 많은 칸을 이동하려면 어떤 경로를 통하여 움직여야 하는지 구하여라.

입력

첫째 줄에 대나무 숲의 크기 n(1 ≤ n ≤ 500)이 주어진다. 그리고 둘째 줄부터 n+1번째 줄까지 대나무 숲의 정보가 주어진다. 대나무 숲의 정보는 공백을 사이로 두고 각 지역의 대나무의 양이 정수 값으로 주어진다. 대나무의 양은 1,000,000보다 작거나 같은 자연수이다.

출력

첫째 줄에는 판다가 이동할 수 있는 칸의 수의 최댓값을 출력한다.

👉 생각

• 완전 탐색으로 모든 칸에서 깊이 우선 탐색을 돌리는데, 이전에 방문했던 곳이 있으면 그 부분은 DFS를 돌리지 않고 res만 갱신을 해주었다.
• 나름 시간초과를 염두해 두었는데도 시간 초과가 발생했다.

from collections import deque
import sys

def dfs(i,j):
    stack = deque([])
    stack.append((i,j,1))
    res = 0
    while stack:
        x, y, cnt = stack.pop()
        res = cnt if cnt > res else res
        for dx, dy in [(-1,0), (1,0), (0,-1),(0,1)]:
            nx, ny = x + dx, y + dy
            if 0 <= nx < n and 0 <= ny < n and arr[nx][ny] > arr[x][y]:
                if visited[nx][ny]:
                    res = cnt + visited[nx][ny] if cnt + visited[nx][ny] > res else res
                else:
                    stack.append((nx, ny, cnt+1))

    visited[i][j] = res
    return res

n = int(input())
arr = [list(map(int, sys.stdin.readline().split())) for _ in range(n)]
visited = [[0] * n for _ in range(n)]
answer = 0
for i in range(n):
    for j in range(n):
        tmp = dfs(i,j)
        answer = tmp if tmp > answer else answer
print(answer)

• 이 문제는 DP + DFS를 같이 써야 시간초과를 통과할 수 있다.
• 재귀를 통해 DFS 탐색을 하면서 방문하지 않은 곳의 제일 깊이 까지 들어간 후, 이전 방문했던 값 + 1과 현재 방문 값 중 큰 값을 선택하고 반환을 해준다.

import sys
sys.setrecursionlimit(1000000)

def dfs(x,y):
    if visited[x][y] == -1:
        visited[x][y] = 0

        for dx, dy in [(-1,0), (1,0), (0,-1),(0,1)]:
            nx, ny = x + dx, y + dy
            if 0 <= nx < n and 0 <= ny < n and arr[nx][ny] > arr[x][y]:
                visited[x][y] = max(visited[x][y], dfs(nx, ny))

    return visited[x][y] + 1

n = int(input())
arr = [list(map(int, sys.stdin.readline().split())) for _ in range(n)]
visited = [[-1] * n for _ in range(n)]
answer = 0
for i in range(n):
    for j in range(n):
        answer = max(answer, dfs(i,j))
print(answer)