[프로그래머스] 무인도 여행 (JAVA)

유존돌돌이·2023년 1월 26일
0

Programmers

목록 보기
160/167
post-thumbnail

문제 설명

지도를 나타내는 문자열 배열 maps가 매개변수로 주어질 때, 각 섬에서 최대 며칠씩 머무를 수 있는지 배열에 오름차순으로 담아 return 하는 solution 함수를 완성해주세요. 만약 지낼 수 있는 무인도가 없다면 -1을 배열에 담아 return 해주세요.

제한사항

3 ≤ maps의 길이 ≤ 100
3 ≤ maps[i]의 길이 ≤ 100
maps[i]는 'X' 또는 1 과 9 사이의 자연수로 이루어진 문자열입니다.
지도는 직사각형 형태입니다.

Code

import java.util.*;
class Solution {
    public int[] solution(String[] maps) {
        boolean[][] visit = new boolean[maps.length][maps[0].length()];
		List<Integer> list = new ArrayList<>();
		
		for(int i=0 ; i<maps.length ; i++) {
			int sum = 0;
			for(int j=0 ; j<maps[i].length() ; j++) {
				sum = getTerritory(i, j, visit, maps);
				if(sum>0) {
					list.add(sum);
				}
			}
		}
		Collections.sort(list);
		return list.isEmpty()?new int[]{-1}:list.stream().mapToInt(Integer::intValue).toArray();
	}
	
	public int getTerritory(int i, int j, boolean[][] visit, String[] maps) {
		if(i<0 || j<0 || i>=visit.length || j>=visit[0].length || visit[i][j] || maps[i].charAt(j)=='X') {
			return 0;
		}
		visit[i][j] = true;
		return (maps[i].charAt(j)-'0')
			+ getTerritory(i-1,j,visit,maps)
			+ getTerritory(i+1,j,visit,maps)
			+ getTerritory(i,j-1,visit,maps)
			+ getTerritory(i,j+1,visit,maps);
	}
}

0개의 댓글