[백준/C++] 1937 번 : 욕심쟁이 판다 풀이

1937번 : 욕심쟁이 판다

문제 링크 : www.acmicpc.net/problem/1937

1937번: 욕심쟁이 판다

---

문제

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

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

 

입력

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

 

출력

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

---

풀이1

#include <iostream>
#include <queue>
#include <cstring>
#define MAX 501

using namespace std;

int n, map[MAX][MAX], ans;
bool visited[MAX][MAX];
int sum[MAX][MAX]; // 판다 경로 저장

const int dy[4] = {-1,1,0,0};
const int dx[4] = {0,0,-1,1};

int max(int a, int b){
    return a > b ? a : b;
}

void bfs(int init_y, int init_x){
    memset(visited, false, sizeof(visited));
    memset(sum, 0, sizeof(sum));
    visited[init_y][init_x] = true;
    
    queue<pair<int,int>> que;
    que.push(make_pair(init_y, init_x));
     
    while(!que.empty()){
        int y = que.front().first;
        int x = que.front().second;
        que.pop();
        
        for(int i=0; i<4; i++){
            int ny = y + dy[i];
            int nx = x + dx[i];
            
            if(ny<0 || nx<0 || ny>=n || nx>=n) continue;
            if(!visited[ny][nx] && map[ny][nx]>map[y][x]){
                visited[ny][nx] = true;
                sum[ny][nx] = sum[y][x]+1;
                ans = max(ans, sum[ny][nx]);
                que.push(make_pair(ny, nx));
            }
        }
        
    }
}

int main(void){
    
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    
    cin >> n;
    
    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            cin >> map[i][j];
        }
    }
  
    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            bfs(i,j);
        }
    }
    
    cout << ans+1 << '\n';
    return 0;
}

시간 초과

완탐 BFS 방법으로 풀었더니 답은 나오지만 시간 초과 오류가 났다 ㅠㅠ

 

 

풀이2

#include <iostream>
#include <cstring>
#define MAX 501

using namespace std;

int n, map[MAX][MAX], ans;
bool visited[MAX][MAX];
int sum[MAX][MAX]; // 판다 경로 저장

const int dy[4] = {-1,1,0,0};
const int dx[4] = {0,0,-1,1};

int max(int a, int b){
    return a > b ? a : b;
}

void dfs(int y, int x){
    visited[y][x] = true;
    
    for(int i=0; i<4; i++){
        int ny = y + dy[i];
        int nx = x + dx[i];
        
        if(ny<0 || nx<0 || ny>=n || nx>=n) continue;
        if(!visited[ny][nx] && map[ny][nx]>map[y][x]){
            sum[ny][nx] = sum[y][x]+1;
            ans = max(ans, sum[ny][nx]);
            dfs(ny,nx);
        }
    }
}

int main(void){
    
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    
    cin >> n;
    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            cin >> map[i][j];
        }
    }
    
    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            memset(visited, false, sizeof(visited));
            memset(sum, 0, sizeof(sum));
            dfs(i, j);
        }
    }
    
    cout << ans+1 << '\n';
    
    return 0;
}

(또) 시간 초과

완탐 BFS 방법이 문제인가 싶어서 완탐DFS 방법으로 바꾸어 풀어보았다. 역시나 답은 잘 나오지만 시간 초과 오류가 났다..

 

 

풀이3

#include <iostream>
#define MAX 501

using namespace std;

int n, map[MAX][MAX], ans;
int sum[MAX][MAX]; // 판다 경로 저장

const int dy[4] = {-1,1,0,0};
const int dx[4] = {0,0,-1,1};

int max(int a, int b){
    return a > b ? a : b;
}

int dfs(int y, int x){
    if(sum[y][x]) return sum[y][x];
    
    sum[y][x] = 1;
    for(int i=0; i<4; i++){
        int ny = y + dy[i];
        int nx = x + dx[i];
        
        if(ny<0 || nx<0 || ny>=n || nx>=n) continue;
        if(map[ny][nx]>map[y][x]){
            sum[y][x] = max(sum[y][x], dfs(ny,nx)+1);
        }
    }
    return sum[y][x];
}

int main(void){
    
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    
    cin >> n;
    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            cin >> map[i][j];
        }
    }
    
    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            ans = max(ans, dfs(i, j));
        }
    }
    
    cout << ans << '\n';
    
    return 0;
}

 

BFS, DFS 가 문제가 아니라 '완전탐색' 이 문제였다. 이 문제는 DP 로 해결해야 한다.

배열 sum[y][x] 에는 판다가 (y,x) 좌표에 있을 때 최대로 살 수 있는 날짜 를 의미한다.

if(sum[y][x]) return sum[y][x];

이미 값이 있는 좌표는 더 이상 방문할 필요가 없으므로 리턴해준다.

값이 있는 경우는 해당 좌표를 1로 설정하고 탐색을 진행해야 한다.

sum[y][x] = 1;

if(map[ny][nx]>map[y][x]){
      sum[y][x] = max(sum[y][x], dfs(ny,nx)+1);
}

 

풀이 인증