[동적 계획법] 백준 1937 :: 욕심쟁이 판다

욕심쟁이 판다 성공

시간 제한	메모리 제한	제출	정답	맞은 사람	정답 비율
2 초	256 MB	15482	4798	3020	29.198%

문제

n*n의 크기의 대나무 숲이 있다. 욕심쟁이 판다는 어떤 지역에서 대나무를 먹기 시작한다. 그리고 그 곳의 대나무를 다 먹어 치우면 상, 하, 좌, 우 중 한 곳으로 이동을 한다. 그리고 또 그곳에서 대나무를 먹는다. 그런데 단 조건이 있다. 이 판다는 매우 욕심이 많아서 대나무를 먹고 자리를 옮기면 그 옮긴 지역에 그 전 지역보다 대나무가 많이 있어야 한다. 만약에 그런 지점이 없으면 이 판다는 불만을 가지고 단식 투쟁을 하다가 죽게 된다(-_-)

이 판다의 사육사는 이런 판다를 대나무 숲에 풀어 놓아야 하는데, 어떤 지점에 처음에 풀어 놓아야 하고, 어떤 곳으로 이동을 시켜야 둘 다 소중한 생명이지만 판다가 최대한 오래 살 수 있는지 고민에 빠져 있다. 우리의 임무는 이 사육사를 도와주는 것이다. n*n 크기의 대나무 숲이 주어져 있을 때, 이 판다가 최대한 오래 살려면 어떤 경로를 통하여 움직여야 하는지 구하여라.

입력

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

출력

첫째 줄에는 판다가 최대한 살 수 있는 일수(K)를 출력한다.

예제 입력 1 

4
14 9 12 10
1 11 5 4
7 15 2 13
6 3 16 8

예제 출력 1 

4

     

 문제 풀이

1차 풀이 (  3940 KB,  36 ms)

어느 지점에서 사방을 검사하고, 먹이가 많다면 이동시킨다.

먹이가 많은쪽으로만 이동하므로, 사이클이 생길 걱정은 없다.

외곽인 경우에 주의한다.

#include <bits/stdc++.h>
using namespace std;

const int SIZE = 500;
int cost[SIZE][SIZE];
int cache[SIZE][SIZE], N;

int solve(int x, int y){
        //! base condition.
        if(x < 0) return 0;
        if(y < 0) return 0;
        if(N <= x) return 0;
        if(N <= y) return 0;

        //! memoization.
        if(cache[x][y] != -1) return cache[x][y];

        //! solving.
        //! 외곽을 체크한 뒤, 먹이가 많다면 이동시킨다.
        int ans = 1;
        if(y != 0 && cost[x][y] < cost[x][y-1]) {
                ans = max(ans, solve(x, y-1) + 1);
        }
        if(x != 0 && cost[x][y] < cost[x-1][y]){
                ans = max(ans, solve(x-1, y) + 1);
        }
        if(x != N-1 && cost[x][y] < cost[x+1][y]) {
                ans = max(ans, solve(x+1, y) + 1);
        }
        if(y != N-1 && cost[x][y] < cost[x][y+1]) {
                ans = max(ans, solve(x, y+1) + 1);
        }
        cache[x][y] = ans;
        return ans;
}

int main()
{
        cin.tie(nullptr);
        ios::sync_with_stdio(false);

        //! data input.
        memset(cache, -1, sizeof cache);
        cin >> N;
        for(int i=0; i<N; i++){
                for(int j=0; j<N; j++){
                        cin >> cost[i][j];
                }
        }

        //! get max.
        int ans = 0;
        for(int i=0; i<N; i++){
                for(int j=0; j<N; j++){
                        ans = max(ans, solve(i, j));
                }
        }
        cout << ans;
}