Python 백준 10250번 - ACM호텔

문제

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

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

사이트 참조

방 번호는 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, 세 정수를 포함하고 있으며 각각 호텔의 층 수, 각 층의 방 수, 몇 번째 손님인지를 나타낸다.

문제풀이

오답률이 많아서 생각보다 알고리즘이 어렵나 생각했으나, 생각보다는 수리 연산으로 끝낼 수 있는 단순한 문제였다. 저런 격자를 보면 처음 드는 생각은 이중 리스트를 만들어서 인덱스로 접근하면 되겠다고 생각했으나, 역시 메모리나 속도에서 떨어질 것이라고 생각이 들어서 최대한 수학 연산으로 끝낼 수 없나 생각이 들었다.

힌트는 순서대로 배치할 때, 층 순으로 올라간다는 것이다. 고로 층을 기준으로 N번째 사람을 묶으면 최소한 몇 층인지 알 수 있었다. 여기서 함정이 있는데 아는 것처럼 mod연산을 할 시에 0으로 나누어 떨어지는 부분이 있다는 것, 여기서 많이 틀렸을 것이라고 생각이 들었다. 나는 그래서 나눗셈 mod연산 후 0 이 되는 부분은 예외처리로 따로 빼두고 처리하였다. 아래 변수 floor가 그 부분이다. 이제 호수(ho)를 결정해야 하는데, 처음 호수를 계산하려고 했을 때는 단순히 층을 N으로  나눈 몫을 가지고 + 1만 해주면 되겠다고 생각을 하였는데 생각을 해보면 역시 나누어 떨어지는 가장 꼭대기 층에서 문제가 하나 있는데 이 층에서만 +1을 해주면 안 된다, 직접 표를 그려서 생각하면 편하다.

 

코드

 

import sys
 
testcase = int(sys.stdin.readline().rstrip())
for i in range(testcase):
    H, W, N = map(int, sys.stdin.readline().split())
    floor, ho  = N % H, N // H
    if floor == 0: print(H * 100 + ho)
    else: print((floor)*100 + (ho + 1))