[프로그래머스] 코딩테스트 연습 > 연습문제 > 무인도 여행
메리는 여름을 맞아 무인도로 여행을 가기 위해 지도를 보고 있습니다. 지도에는 바다와 무인도들에 대한 정보가 표시돼 있습니다. 지도는 1 x 1크기의 사각형들로 이루어진 직사각형 격자 형태이며, 격자의 각 칸에는 'X' 또는 1에서 9 사이의 자연수가 적혀있습니다. 지도의 'X'는 바다를 나타내며, 숫자는 무인도를 나타냅니다. 이때, 상, 하, 좌, 우로 연결되는 땅들은 하나의 무인도를 이룹니다. 지도의 각 칸에 적힌 숫자는 식량을 나타내는데, 상, 하, 좌, 우로 연결되는 칸에 적힌 숫자를 모두 합한 값은 해당 무인도에서 최대 며칠동안 머물 수 있는지를 나타냅니다. 어떤 섬으로 놀러 갈지 못 정한 메리는 우선 각 섬에서 최대 며칠씩 머물 수 있는지 알아본 후 놀러갈 섬을 결정하려 합니다.
지도를 나타내는 문자열 배열 maps가 매개변수로 주어질 때, 각 섬에서 최대 며칠씩 머무를 수 있는지 배열에 오름차순으로 담아 return 하는 solution 함수를 완성해주세요. 만약 지낼 수 있는 무인도가 없다면 -1을 배열에 담아 return 해주세요.
maps | result |
---|---|
["X591X","X1X5X","X231X", "1XXX1"] | [1, 1, 27] |
["XXX","XXX","XXX"] | [-1] |
재귀를 사용하여 해결하였다.
import java.util.*;
class Solution {
int cnt;
public int[] solution(String[] maps) {
int[] answer = null;
int[][] map = new int[maps.length][maps[0].length()];
for (int i = 0; i < maps.length; i++) {
for (int j = 0; j < maps[0].length(); j++) {
map[i][j] = maps[i].charAt(j) == 'X' ? 0 : Character.getNumericValue(maps[i].charAt(j));
}
}
ArrayList<Integer> list = new ArrayList<Integer>();
for (int i = 0; i < maps.length; i++) {
for (int j = 0; j < maps[0].length(); j++) {
if (map[i][j] != 0) {
cnt = 0;
dfs(map, i, j);
list.add(cnt);
}
}
}
if(list.size() == 0) {
answer = new int[1];
answer[0] = -1;
} else {
list.sort(Comparator.naturalOrder());
answer = new int[list.size()];
for (int i = 0; i < list.size(); i++)
answer[i] = list.get(i);
}
return answer;
}
public void dfs(int[][] map, int r, int c) {
if (r < 0 || c < 0 || r >= map.length || c >= map[0].length)
return;
if (map[r][c] == 0)
return;
cnt += map[r][c];
map[r][c] = 0;
dfs(map, r + 1, c);
dfs(map, r - 1, c);
dfs(map, r, c + 1);
dfs(map, r, c - 1);
}
}