[프로그래머스/Java] 거리두기 확인하기

문제

https://school.programmers.co.kr/learn/courses/30/lessons/81302

개발자를 희망하는 죠르디가 카카오에 면접을 보러 왔습니다.
코로나 바이러스 감염 예방을 위해 응시자들은 거리를 둬서 대기를 해야하는데 개발 직군 면접인 만큼
아래와 같은 규칙으로 대기실에 거리를 두고 앉도록 안내하고 있습니다.

대기실은 5개이며, 각 대기실은 5x5 크기입니다.
거리두기를 위하여 응시자들 끼리는 맨해튼 거리1가 2 이하로 앉지 말아 주세요.
단 응시자가 앉아있는 자리 사이가 파티션으로 막혀 있을 경우에는 허용합니다.

예를 들어,

5개의 대기실을 본 죠르디는 각 대기실에서 응시자들이 거리두기를 잘 기키고 있는지 알고 싶어졌습니다. 자리에 앉아있는 응시자들의 정보와 대기실 구조를 대기실별로 담은 2차원 문자열 배열 places가 매개변수로 주어집니다. 각 대기실별로 거리두기를 지키고 있으면 1을, 한 명이라도 지키지 않고 있으면 0을 배열에 담아 return 하도록 solution 함수를 완성해 주세요.

---

접근 방법

• 각 대기실의 상태를 3차원 배열 map에 복사한 후, check 함수를 통해 거리두기 규칙을 검사
• 거리두기 규칙이 지켜지면 1, 그렇지 않으면 0을 반환하여 최종 결과를 answer 배열에 저장하고 반환

거리두기check
1. 응시자 : 응시자 상하좌우에 다른 응시자가 있으면 지켜지지 않음
2. 빈 테이블 : 빈 테이블 상하좌우에 응시자 수를 세고, 인접한 응시자가 2명 이상하면 지켜지지 않음
3. 파티션 : 검사를 하지 않고 넘어감

---

코드 구현

import java.util.*;

class Solution {
    static char[][][] map;
    static int[] dx = {0, 0, 1, -1};
    static int[] dy = {1, -1, 0, 0};
    public int[] solution(String[][] places) {
        map = new char[5][5][5];
        int[] answer = {1, 1, 1, 1, 1}; // 응시자가 한명도 없는 경우, 거리두기가 지켜진 것
        
        for(int i = 0; i < 5; i++){
            for(int j = 0; j < 5; j++){
                for(int k = 0; k < 5; k++)
                    map[i][j][k] = places[i][j].charAt(k);
            }
            answer[i] = check(map[i]);
        }
        return answer;
    }
    
    static int check(char[][] m){
        for(int i = 0; i < 5; i++){
            for(int j = 0; j < 5; j++){
                if(m[i][j] == 'X') // 파티션(검사할 필요X)
                    continue;
                else if(m[i][j] == 'P'){ // 응시자
                    for(int k = 0; k < 4; k++){
                        int nx = i + dx[k];
                        int ny = j + dy[k];
                        
                        if(nx >= 0 && ny >= 0 && nx < 5 && ny <5){
                            if(m[nx][ny] == 'P') // 상하좌우에 또다른 응시자 존재
                                return 0;
                        }
                    }
                }
                else { // 빈 테이블일 경우
                    int count = 0; // 인접한 응시자(P)의 수를 세기 위한 변수
                    
                    for(int k = 0; k < 4; k++){
                        int nx = i + dx[k];
                        int ny = j + dy[k];
                        
                        if(nx >= 0 && ny >= 0 && nx < 5 && ny <5){
                            if(m[nx][ny] == 'P') // 상하좌우에 또다른 응시자 존재
                                count++;
                        }
                    }
                    // 빈 테이블과 인접한 응시자가 2명 이상이면 거리두기 실패
                    if(count >= 2)
                        return 0; 
                }
            }
        }
        return 1; // 모든 검사를 통과하면 거리두기 성공
    }
}