[백준] # 1937. 욕심쟁이 판다 (파이썬)

문제 링크

 

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

1937번: 욕심쟁이 판다

 

문제 내용

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

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

 

입력

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

출력

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

 

IDEA

dp라는 배열에 [x,y]좌표에서 시작했을 때 이동할 수 있는 칸의 최댓값을 저장하고, 이를 활용하여 반복적인 계산을 없앴다. 나무의 수를 인덱스로 하고, 그 좌표(x, y)를 저장하는 location이라는 배열을 생성하여 가장 나무가 많은 곳부터 dp를 계산하게 했다.

 

CODE

# 1937. 욕심쟁이 판다 - 골드3
def findValue(x, y):
    # cnt : 상하좌우에 자신 보다 큰 값의 개수
    cnt = 0
    # dfs
    for dx,dy in d:
        nx = x + dx
        ny = y + dy
        if 0 <= nx < n and 0 <= ny < n and forest[x][y] < forest[nx][ny]:
            dp[x][y] = max(dp[x][y],dp[x+dx][y+dy] + 1)
            cnt += 1
    # 주위에서 가장 클 때,
    if cnt == 0:
        dp[x][y] = 1
n = int(input())
forest = []
d = [(0,1),(1,0),(0,-1),(-1,0)]
# dp[x,y] : x,y좌표에서 시작했을 때 이동할 수 있는 칸의 최댓값
dp = [[0] * n for _ in range(n)]

# location에 좌표 저장
location = [[] for _ in range(1000001)]
for i in range(n):
    temp = list(map(int,input().split()))
    forest.append(temp)
    for j in range(n):
        location[temp[j]].append((i,j))
max_value = max(map(max,forest))
# dp 계산
for i in range(max_value,0,-1):
    if len(location[i]) == 0:
        continue
    else:
        for x,y in location[i]:
            findValue(x,y)
print(max(map(max,dp)))

 

사용 알고리즘

DP, DFS

 

풀이 평

2시간 걸려 겨우 풀었다. 알고리즘을 빨리 생각했으면 됐는데 아직 DFS, BFS에 능숙하지 않았기도 했고, DP와 섞인 문제는 처음이라 어려웠다. 그렇지만 재미있었다.