[BOJ(백준) 10250] ACM호텔 - 파이썬 풀이

https://www.acmicpc.net/problem/10250

10250번: ACM 호텔

문제

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 번째 손님에게 배정되어야 하는 방 번호를 출력한다.

---

 문제에서 손님은 엘리베이터를 타는 시간은 상관없고, 엘리베이터에서 내려서 방까지 걸어가는 시간이 짧아야 한다. (같은 거리라면 더 낮은 층수의 방을 손님에게 배정해야 한다.) 호텔이 6층이라면 방이 6층까지 있고 손님이 오면 101,201,301... 601, 102,202... 순으로 방을 줘야 한다. 나는 두 가지 방법으로 풀었는데 1) 반복문 2중으로 사용한 방식과  2) 수학 계산을 사용한 방식으로 풀었다.

 

t = int(input())
for _ in range(t):
    h, w, n = map(int, input().split())
    count = 0
    for i in range(1,w+1):
        if count == n:
            break
        for j in range(1,h+1):
            count += 1
            if count == n:
                if i<10:
                    print(f"{j}0{i}")
                else:
                    print(f"{j}{i}")
                break

 방법 1)은 전체 방의 개수만큼 반복문을 돌리는 방식이다. 각 층의 방의 개수(W)만큼 반복을 돌리고, 그 안에 건물의 층수(H)만큼 반복을 한다. count는 몇 번째 손님인지 세는 함수인데 count와 인풋 값(t)이 같으면 그 방의 번호를 출력하고 이중 반복문을 종료한다.

 

t = int(input())
for _ in range(t):
    h, w, n = map(int, input().split())
    y = h if n % h == 0 else n % h
    x = n // h if n % h == 0 else n // h + 1

    if x < 10:
        print(f"{y}0{x}")

    else:
        print(f"{y}{x}")

 방법 2)는 수학 계산으로 방 번호를 계산한다. 방 번호는 YXX 또는 YYXX형식인데, Y(또는 YY)는 방의 층수를 나타내고 XX는 엘리베이터부터 순서대로 센 방의 번호를 나타낸다. 호텔이 H=6, W=12라면, 첫 번째부터 여섯 번째 손님까지는 각층의 1번 방에 머물 수 있고, 일곱 번째 손님부터 열두 번째 손님까지는 각 층의 2번 방에 머물 수 있다. 만약 열 번째 손님이 묵을 방을 알고 싶다면, 10을 호텔의 층수인 6으로 나눈다. 여기서 Y는 10을 6으로 나눈 몫에 1을 더한 값이고, XX는 10을 6으로 나눈 나머지 값인 4, 즉 04가 된다.

 코드로 짤 때 주의할 것은, 손님의 번호를 호텔의 층수로 나눌 때 그 값이 나누어 떨어진다면, Y(또는 YY)=H로 Y가 호텔 최고층의 높이가 되고, XX는 N을 호텔의 층수로 나눈 몫이 된다.

 

 두 가지 방법을 썼을 때 시간은 다음과 같았다. 만약 호텔의 크기가 아주 크고, 손님의 번호도 아주 크다면 1번 방법은 더 많은 시간이 걸릴 것이다. 비슷한 문제를 다시 풀어야 한다면 가능한 2번 방법으로 풀고 싶다.(2번으로 풀기 어려운 문제라면 1번 방법으로 풀어야겠지만 ㅎㅎ)