[백준 1937 - C++] 욕심쟁이 판다 : DFS + DP

www.acmicpc.net/problem/1937

1937번: 욕심쟁이 판다

문제

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

 

풀이

이 문제는 DFS와 DP를 함께 사용해야하는 문제이다. 처음엔 그냥 DFS만 사용해서 시간초과로 틀렸다ㅠㅠ

그래서 dfs를 진행하면서 구해진 값을 keep이라는 배열을 하나 만들어서 계산하면서 살 수 있는 일 수를 저장하고 다시 사용하여, 동일한 계산이 반복되는 것을 예방할 수 있다.

 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
using namespace std;

int map[501][501];
int keep[501][501];

int R[4] = {1, -1, 0, 0};
int C[4] = {0, 0, 1, -1};


int dfs(int n, int r, int c);

int main(void){
    int n;
    int max_d = 0;
    int temp;
    
    scanf("%d", &n);
    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            scanf("%d", &map[i][j]);
        }
    }
    
    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            
            temp = dfs(n, i, j);
            //printf("%d ", temp);
            if(temp > max_d)
                max_d = temp;
        }
        //printf("\n");
    }
    printf("%d\n", max_d);
    
}

int dfs(int n, int r, int c){
    int next_r, next_c;
    int temp;
    
    if(keep[r][c])
        return keep[r][c];
    
    keep[r][c] = 1;

    
    
    for(int i=0; i<4; i++){
        next_r = r + R[i];
        next_c = c + C[i];
        if(next_r < 0 || next_r >=n || next_c < 0 || next_c >=n)
            continue;
        if(map[next_r][next_c] > map[r][c]){
            temp = dfs(n, next_r, next_c) + 1;
            if(keep[r][c] < temp)
                keep[r][c] = temp;
        }
    }
    
    return keep[r][c];
}