[Baekjoon/JAVA] 1652번 - 누울 자리를 찾아라

Baekjoon Online Judge

문제로 이동

 

문제

일 년 동안 세계일주를 하던 영식이는 여행을 하다 너무 피곤해서 근처에 있는 코레스코 콘도에서 하룻밤 잠을 자기로 하고 방을 잡았다.

코레스코 콘도에 있는 방은 NxN의 정사각형모양으로 생겼다. 방 안에는 옮길 수 없는 짐들이 이것저것 많이 있어서 영식이의 누울 자리를 차지하고 있었다. 영식이는 이 열악한 환경에서 누울 수 있는 자리를 찾아야 한다. 영식이가 누울 수 있는 자리에는 조건이 있다. 똑바로 연속해서 2칸 이상의 빈 칸이 존재하면 그 곳에 몸을 양 옆으로 쭉 뻗으면서 누울 수 있다. 가로로 누울 수도 있고 세로로 누울 수도 있다. 누울 때는 무조건 몸을 쭉 뻗기 때문에 반드시 벽이나 짐에 닿게 된다. (중간에 어정쩡하게 눕는 경우가 없다.)

만약 방의 구조가 위의 그림처럼 생겼다면, 가로로 누울 수 있는 자리는 5개이고, 세로로 누울 수 있는 자리는 4개 이다. 방의 크기 N과 방의 구조가 주어졌을 때, 가로로 누울 수 있는 자리와 세로로 누울 수 있는 자리의 수를 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 방의 크기 N이 주어진다. N은 1이상 100이하의 정수이다. 그 다음 N줄에 걸쳐 N개의 문자가 들어오는데 '.'은 아무것도 없는 곳을 의미하고, 'X'는 짐이 있는 곳을 의미한다.

 

출력

첫째 줄에 가로로 누울 수 있는 자리와 세로로 누울 수 있는 자리의 개수를 출력한다.

 

---

예제 입력	예제 출력
5 ....X ..XX. ..... .XX.. X....	5 4

---

풀이

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);

        int N = in.nextInt(); // 방의 크기 N
        String[] room = new String[101];

        // 입력
        for (int i = 0; i < N; i++)
            room[i] = in.next();

        int garo = garoSero(N, room, false); // 가로로 누울 수 있는 자리의 개수
        int sero = garoSero(N, room, true); // 세로로 누울 수 있는 자리의 개수

        // 출력
        System.out.println(garo + " " + sero);
    }

    // 누울 수 있는 자리의 개수
    public static int garoSero(int N, String[] room, boolean isSero) {
        int num = 0;

        for (int i = 0; i < N; i++) {
            int count = 0; // 연속해서 비어있는 칸

            for (int j = 0; j < N; j++) {
                char c = room[i].charAt(j);

                if (isSero) // 세로칸을 탐색할 때
                    c = room[j].charAt(i);

                // 해당 칸이 비어있는가?
                if (c == '.') count++;
                else if (c == 'X') {
                    // 2칸 이상의 빈칸이 존재할 때
                    if (count >= 2) num++;

                    count = 0;
                }
            }

            // 2칸 이상의 빈칸이 존재할 때
            if (count >= 2) num++;
        }

        return num;
    }
}

가로/ 세로로 누울 수 있는 자리의 개수를 구하는 함수를 구현한 뒤 그 값을 출력했다.

 

    // 누울 수 있는 자리의 개수
    public static int garoSero(int N, String[] room, boolean isSero) {
        int num = 0;

        for (int i = 0; i < N; i++) {
            int count = 0; // 연속해서 비어있는 칸

            for (int j = 0; j < N; j++) {
                char c = room[i].charAt(j);

                if (isSero) // 세로칸을 탐색할 때
                    c = room[j].charAt(i);

                // 해당 칸이 비어있는가?
                if (c == '.') count++;
                else if (c == 'X') {
                    // 2칸 이상의 빈칸이 존재할 때
                    if (count >= 2) num++;

                    count = 0;
                }
            }

            // 2칸 이상의 빈칸이 존재할 때
            if (count >= 2) num++;
        }

        return num;
    }

isSero가 true면 세로로 누울 수 있는 자리의 수를 false면 가로로 누울 수 있는 자리의 개수를 구하는 함수이다. 

각 칸을 하나하나 살펴본 뒤 빈칸이면 count 값을 증가시키고, 빈칸이 아니면 count를 0으로 초기화하는 방식으로 칸이 연속해서 비어있는지를 확인했다.