시간제한 : 1초
메모리제한 : 256MB
문제 :
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, 세 정수를 포함하고 있으며 각각 호텔의 층 수, 각 층의 방 수, 몇 번째 손님인지를 나타낸다(1 ≤ H, W ≤ 99, 1 ≤ N ≤ H × W).
출력 :
프로그램은 표준 출력에 출력한다. 각 테스트 데이터마다 정확히 한 행을 출력하는데, 내용은 N 번째 손님에게 배정되어야 하는 방 번호를 출력한다.
풀이
호텔의 방 배정은 먼저 온 손님의 순서대로 배정되므로, N번째 손님에게 배정되어야 하는 방 번호를 출력하기 위해서는 호텔의 방 배정 규칙을 파악해야 한다.
1) 방은 입구(엘레베이터 바로 앞에 있음)로부터 이동 거리가 짧은 순서대로 배정된다. 이때, 엘레베이터는 이동 거리에 포함되지 않는다.
2) 입구로부터 이동 거리가 동일한 방이라면, 낮은 층이 우선 배정된다.
그러므로 방은 XX01호이 아래층부터 차례대로 배정되며, XX01호가 만실이라면 XX02호가 아래층부터 차례대로 배정된다.
이를 반복문을 통해 표현할 수 있다.
먼저, 방의 호수를 XXYY라고 할 때, XX가 1씩 높아질때마다 방문한 손님의 차례N에 -1씩 해준다.
주어진 층 수만큼 반복이 완료되었다면, 다시 다음 호수(YY+1)로 넘어가서 위의 과정을 반복한다.
이렇게 반복문을 수행하다가 N이 0이 되면 반복문을 종료하고 그 때의 방 호수를 출력하면 된다.
T = int(input())
hotel_list = []
for _ in range(T):
H, W, N = map(int, input().split())
tuple_room = H, W, N
hotel_list.append(tuple_room)
def room_num(height, width, number):
number = number
for i in range(width):
for j in range(height):
number -= 1
if number == 0:
return str(j+1)+str(i+1).zfill(2)
for H, W, N in hotel_list:
print(room_num(H, W, N))
'코딩 연습' 카테고리의 다른 글
[백준 2839] 설탕 배달 / 파이썬 (0) | 2022.09.14 |
---|---|
[백준 2775] 부녀회장이 될테야 / 파이썬 (0) | 2022.09.03 |
[백준 2869] 달팽이는 올라가고 싶다 / 파이썬 (0) | 2022.09.01 |
[백준 1193] 분수찾기 / 파이썬 (1) | 2022.08.31 |
[백준 2292] 벌집 / 파이썬 (0) | 2022.08.30 |