[백준] 1937 - 욕심쟁이 판다 (java)

HaYeong Jang·2021년 1월 23일
0

[백준] 알고리즘

목록 보기
17/62
post-thumbnail
post-custom-banner

문제

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

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

입력

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

출력

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

예제 입력

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

예제 출력

4

코드

import java.io.*;
import java.util.*;

public class Main {
    static int n;
    static int[][] map;
    static int[][] day;
    static int[] dx = {1, -1, 0, 0};
    static int[] dy = {0, 0, 1, -1};

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringTokenizer st = new StringTokenizer(br.readLine());

        n = Integer.parseInt(st.nextToken());
        map = new int[n][n];
        day = new int[n][n];

        for (int i = 0; i < n; i++) {
            st = new StringTokenizer(br.readLine());
            for (int j = 0; j < n; j++) {
                map[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        int max_day = 0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                max_day = Math.max(dfs(i, j), max_day);
            }
        }
        bw.write(max_day + "\n");

        br.close();
        bw.flush();
        bw.close();
    }

    public static int dfs(int x, int y) {
        if (day[x][y] != 0) {   // 이미 계산한 곳이면 return
            return day[x][y];
        }

        int cnt = 1;
        for (int i = 0; i < 4; i++) {   // 상하좌우 반복
            int nx = x + dx[i];
            int ny = y + dy[i];

            if (nx >= 0 && nx < n && ny >= 0 && ny < n) {   // 대나무 숲을 벗어나지 않는지
                if (map[x][y] < map[x + dx[i]][y + dy[i]]) {    // 옮긴 지역이 전 지역보다 많을 경우
                    cnt = Math.max(dfs(nx, ny) + 1, cnt);
                    day[x][y] = cnt;
                }
            }
        }

        return cnt;
    }
}

정리


3달 전에 풀었던 문제인데 BufferedReader, BufferedWriter를 사용하여 다시 풀었더니 메모리와 시간이 눈에 띄게 줄었다.
이 문제는 DFS + DP를 사용하여 풀어야 했다.

profile
기억하기 위해 기록하는 개발로그👣
post-custom-banner

0개의 댓글