[백준]#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

풀이

이 문제는 dp(다이나믹 프로그래밍) 알고리즘을 이용해서 풀 수 있었다.

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.PriorityQueue;

public class Main {
	static int[][] map;
	static int[][] dp;
	static int[] dx = {-1, 1, 0, 0};
	static int[] dy = {0, 0, -1, 1};
	static int max = 0;
	static int N;

    public static void main(String[] args) throws Exception{
	    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

	    N = Integer.parseInt(br.readLine());
	    map = new int[N][N];
	    dp = new int[N][N];
		  PriorityQueue<Pair> pq = new PriorityQueue<>();

	    for(int i=0; i<N; i++) {
	        String[] input = br.readLine().split(" ");
	        for(int j=0; j<N; j++) {
	            map[i][j] = Integer.parseInt(input[j]);
	            pq.add(new Pair(i, j, map[i][j]));
            }
        }

	    while(!pq.isEmpty()) {
	    	Pair temp = pq.poll();

	    	for(int i=0; i<4; i++) {
	    		int nx = temp.x+dx[i];
	    		int ny = temp.y+dy[i];

	    		if(nx<0 || nx>=N || ny<0 || ny>=N || map[temp.x][temp.y]>=map[nx][ny] || dp[nx][ny]>=dp[temp.x][temp.y]+1) continue;

	    		dp[nx][ny] = dp[temp.x][temp.y]+1;

	    		max = Math.max(dp[nx][ny], max);
	    	}
		}

	    System.out.println(max+1);
    }

    public static class Pair implements Comparable<Pair>{
    	int x;
    	int y;
    	int tree;

    	public Pair(int x, int y, int tree) {
    		this.x = x;
    		this.y = y;
    		this.tree = tree;
		}

		public int compareTo(Pair p) {
    		return this.tree > p.tree ? 1 : -1;
		}
	}
}