문제
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 번째 손님에게 배정되어야 하는 방 번호를 출력한다.
구상
방 배정 방식을 요약하면 왼쪽 아래 101호부터 각 층 1호실을 다 채우고, 2호실을 채우고, ... W호실까지 채운다.
H = 6, W = 12인 호텔에 10번째로 방문한 손님이 배정된 방식을 생각해보자.
6번째 손님까지는 각 층 1호실에 배정되고 다음 손님부터는 각 층 2호실에 배치된다. 7번째 손님은 102호, 8번째 손님은 202호, 9번째 손님은 302호, 10번째 손님은 402호에 묵게 된다.
N번째로 방문한 손님은 몇 층에 배정되는가?
N을 H로 나눈 나머지에 해당하는 층에 배정된다. 층을 표현하는 변수를 Y라 하면
Y = N % H 이다.
그럼 몇 호실에 배정되는가?
N을 X로 나눈 몫(이미 채워진 호실)에 1을 더한 방에 배정된다. 호수를 표현하는 변수를 X라 하면
X = N // H + 1 이다.
이러면 예제 입력에 있는 경우는 올바르게 배정할 수 있다.
하지만 N % H = 0일 경우(꼭대기 층인 경우) 문제가 생긴다. 예를 들어 H = 3, W = 10, N = 9라 하자. 이 손님은 303호에 배정받아야 하는데 위 식대로라면 Y = 0, X = 4가 된다. 위 경우는 한 줄이 이미 가득 찬 상태를 가정하고 있기 때문에 X에 1을 추가해 방을 한 칸 옆으로 옮긴다. 꼭대기 층이므로 Y = H로 해도 무방하다.
즉 이 경우엔 Y = H, X = N // H 로 해야 한다.
한 가지 더. 아홉 번째 방까지는 '0X'호라고 표시해야 한다. 예를 들어 1층 6번 째 방은 106호로 표시한다. 중간에 0을 넣어야 한다.
출력만 합쳐서 하면 되니 층 수와 방 번호를 따로 구했다. Y와 X를 문자열로 형변환하고 합쳐서 출력하면 된다.
코드
def func(h,w,n):
if n % h == 0: #꼭대기 층일 때
y = h
x = n // h
else:
y = n % h
x = n // h + 1
return x, y
test = int(input())
for i in range(test):
h, w, n = map(int, (input().split()))
x, y = func(h,w,n)
if x // 10 == 0: #아홉 번째 방까지는 중간에 0 추가
print(str(y)+'0'+str(x))
else:
print(str(y)+str(x))
'프로그래밍 > Baekjoon' 카테고리의 다른 글
(파이썬) 백준 1085. 직사각형에서 탈출 (0) | 2020.09.06 |
---|---|
(파이썬) 백준 2665. 부녀회장이 될테야 (0) | 2020.09.04 |
(파이썬) 백준 2869. 달팽이는 올라가고 싶다 (0) | 2020.09.02 |
(파이썬) 백준 1193. 분수찾기 (0) | 2020.08.31 |
(파이썬)백준 1065. 한수 (0) | 2020.08.29 |