[10250번] ACM 호텔 ACM Hotel

수학, 구현, 211101

문제

ACM 호텔 매니저 지우는 손님이 도착하는 대로 빈 방을 배정하고 있다. 고객 설문조사에 따르면 손님들은 호텔 정문으로부터 걸어서 가장 짧은 거리에 있는 방을 선호한다고 한다. 여러분은 지우를 도와 줄 프로그램을 작성하고자 한다. 즉 설문조사 결과 대로 호텔 정문으로부터 걷는 거리가 가장 짧도록 방을 배정하는 프로그램을 작성하고자 한다.

문제를 단순화하기 위해서 호텔은 직사각형 모양이라고 가정하자. 각 층에 W 개의 방이 있는 H 층 건물이라고 가정하자 (1 ≤ H, W ≤ 99). 그리고 엘리베이터는 가장 왼쪽에 있다고 가정하자(그림 1 참고). 이런 형태의 호텔을 H × W 형태 호텔이라고 부른다. 호텔 정문은 일층 엘리베이터 바로 앞에 있는데, 정문에서 엘리베이터까지의 거리는 무시한다. 또 모든 인접한 두 방 사이의 거리는 같은 거리(거리 1)라고 가정하고 호텔의 정면 쪽에만 방이 있다고 가정한다.

그림 1. H = 6 이고 W = 12 인 H × W 호텔을 간략하게 나타낸 그림

방 번호는 YXX 나 YYXX 형태인데 여기서 Y 나 YY 는 층 수를 나타내고 XX 는 엘리베이터에서부터 세었을 때의 번호를 나타낸다. 즉, 그림 1 에서 빗금으로 표시한 방은 305 호가 된다.

손님은 엘리베이터를 타고 이동하는 거리는 신경 쓰지 않는다. 다만 걷는 거리가 같을 때에는 아래층의 방을 더 선호한다. 예를 들면 102 호 방보다는 301 호 방을 더 선호하는데, 102 호는 거리 2 만큼 걸어야 하지만 301 호는 거리 1 만큼만 걸으면 되기 때문이다. 같은 이유로 102 호보다 2101 호를 더 선호한다.

여러분이 작성할 프로그램은 초기에 모든 방이 비어있다고 가정하에 이 정책에 따라 N 번째로 도착한 손님에게 배정될 방 번호를 계산하는 프로그램이다. 첫 번째 손님은 101 호, 두 번째 손님은 201 호 등과 같이 배정한다. 그림 1 의 경우를 예로 들면, H = 6이므로 10 번째 손님은 402 호에 배정해야 한다.

입력

프로그램은 표준 입력에서 입력 데이터를 받는다. 프로그램의 입력은 T 개의 테스트 데이터로 이루어져 있는데 T 는 입력의 맨 첫 줄에 주어진다. 각 테스트 데이터는 한 행으로서 H, W, N, 세 정수를 포함하고 있으며 각각 호텔의 층 수, 각 층의 방 수, 몇 번째 손님인지를 나타낸다(1 ≤ H, W ≤ 99, 1 ≤ N ≤ H × W).

출력

프로그램은 표준 출력에 출력한다. 각 테스트 데이터마다 정확히 한 행을 출력하는데, 내용은 N 번째 손님에게 배정되어야 하는 방 번호를 출력한다.

예제 입력 1

Plain Text
복사

예제 출력 1

Plain Text
복사

My solution

브론즈 3 문제치고는 오래 걸렸다. 한 시간정도 헤맸는데 이유는 0으로 채워진 층과 호수만큼의 배열을 만들어놓고 맨 앞의 인덱스부터 N만큼 1로 채워준 다음 N이 되면 해당 인덱스를 가져오는 방식으로 문제를 풀었기 때문이다. 이렇게 했더니 typeError가 나고 삼중 for문도 만들어졌다.

그래서 처음에 하려고 했던 몫과 나머지를 이용했다.

맨 처음에 N을 H로 나눈 나머지가 층수가 된다는 사실을 알게 되고 그대로 문제를 풀었다가 틀려서 위의 방법을 시도해봤던 것인데, 틀렸던 이유는 N과 H가 똑같은 5라고 했을 때 N % H = 0이 되어 5층이 아니라 0층에 지정되어진 것으로 계산되기 때문이었다.

그런데 이 엣지 케이스를 처리해주면 되는 것이었다. N % H === 0 ? H : N % H;로 하면 원하는 층수를 얻을 수 있었다.

const floor = N % H === 0 ? H : N % H;
JavaScript
복사

그리고 N / H 로 몫을 구하는 것도 필요한데, 이것으로 해당 층의 몇 호수에 머무는지에 대해 구할 수 있었다. 물론 js에서 N / H로 몫을 구하면 소수점이 생기곤 한다. 이 경우 생각해보면 N이 4일 때 H가 6이면 4 / 6 = 0.XXX가 되지만 실제로 호수는 맨 처음 호수에 해당한다. 따라서 0으로 만들어주는 것(내림)이 필요했고 일반화하면 Math.ceil()메서드로 감싸서 내림해주면 올바른 호수가 구해졌다.

const num = Math.ceil(N / H).toString();
JavaScript
복사

이렇게 구하고 나서 YXX, YYXX의 형식을 맞춰주기 위해서 fillZero(str)라는 함수를 만들었다. 따로 width를 지정해주지 않고 2로 고정해주었다.

function fillZero(str) {
    return str.length >= 2
        ? str
        : new Array(2 - str.length + 1).join("0") + str;
}
JavaScript
복사

개발을 하면서 날짜를 구하는 경우에 fillZero() 함수를 구현할 일이 있었는데 오랜만에 구현해보니 조금 헷갈렸다. 만약 원하는 width만큼의 숫자가 안 들어오면 대신 0으로 채워주어야 하는데 그것을 어떻게 할 것인가를 생각해보면 new Array(2 - str.length + 1)로 0이 필요한 개수의 +1만큼으로 Array를 만들어주었다.

뒤에 join("0")메서드를 사용하기 때문에 세 개의 empty item을 2개의 "0"으로 연결시켜주면서 "00"을 만들게 된다. 그래서 2 - str.length가 아니라 2 - str.length +1을 해준 것이다.

소스코드

function solution(i) {
    let answer = [];
    const input = i.toString().trim().split("\n");
    input.shift();

    function fillZero(str) {
        return str.length >= 2
            ? str
            : new Array(2 - str.length + 1).join("0") + str;
    }

    for (let x of input) {
        let arr = x.split(" ").map((v) => parseInt(v));
        [H, W, N] = arr;
        const floor = N % H === 0 ? H : N % H;
        const num = Math.ceil(N / H).toString();
        answer.push(floor + fillZero(num));
    }
    console.log(answer.join("\n"));

    return answer.join("\n");
}

test("solution", () => {
    expect(solution("2\n6 12 10\n30 50 72\n1 10 10")).toStrictEqual(
        "402\n1203\n110"
    );
});
JavaScript
복사