알고리즘(C++) / 백준 1937 : 욕심쟁이 판다

1937

http://boj.kr/1937

1937번: 욕심쟁이 판다

 

문제

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

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

 

입력

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

 

출력

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

 

코드

//1937 욕심쟁이 판다
#include <iostream>
#include <cstring>
using namespace std;
int n;
int denamu[501][501]; //대나무 숲
int visited[501][501]; //이동 거리
int move_panda = 0;
int max_move = 0;
int nx[4] = { 1, 0, -1, 0 }, ny[4] = { 0, 1, 0, -1 };

int DFS(int x, int y) {
	if (visited[x][y]) //방문했던 위치
		return visited[x][y];

	visited[x][y] = 1;
	move_panda = 0;

	for (int i = 0; i < 4; i++) {
		int next_x = x + nx[i];
		int next_y = y + ny[i]; 
		//이동할 위치
		
		if (next_x < 0 || next_y < 0 || next_x >= n || next_y >= n)
			continue;
		//벗어난 위치

		if (denamu[x][y] < denamu[next_x][next_y]) {
			move_panda = DFS(next_x, next_y) + 1;
			if (visited[x][y] < move_panda)
				visited[x][y] = move_panda;
		}
		//이전 위치보다 대나무 많은 곳
	}
	
	return visited[x][y];
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	cin >> n;

	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			cin >> denamu[i][j];
		}
	}

	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			int result = DFS(i, j);
			if (max_move < result) {
				max_move = result;
			}
		}
	}

	cout << max_move << "\n";

	return 0;
}

 

설명

denamu 배열에 대나무 숲의 정보를 입력받는다.

모든 배열을 방문하여 최대로 살 수 있는 일수를 깊이 우선 탐색 DFS로 구하게 된다.

DFS는 이미 전에 방문했던 곳은 바로 return한다.

판다가 이동할 수 있는 곳은 위,아래,왼쪽,오른쪽이고 배열의 범위 밖을 벗어나면 continue한다. 그리고 이동할 위치의 대나무 숲이 현재 위치하고 있는 대나무 숲보다 나무가 많아야지만 이동이 가능하다. 

visited 배열에는 그 위치에서 최대한 살 수 있는 일 수를 저장하고 있다. 

 

 

 

결과

 

고찰

문제는 풀었지만 시간초과로 인하여 애를 먹었다.

구글링한 결과 DP와 DFS를 같이 사용하여 문제를 풀었지만 DP를 아직 잘 모르기 때문에 나중에 다시 풀어봐야할 거 같다. 

시간 초과를 해결하기 위해서 if문을 사용하여 조건을 추가해줄 수 있었다.

 

난이도

Gold 3