[Python] 백준 - 10250번

SMongS·2022년 12월 3일
1

CodingTest

목록 보기
30/49
post-custom-banner

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

예제 입력 1

2
6 12 10
30 50 72

예제 출력 1

402
1203

코드

import sys, math

T = int(sys.stdin.readline().rstrip())

for i in range(T):
  H, W, N = map(int, sys.stdin.readline().split())
  y = N % H
  x = str(math.ceil(N / H))

  if N % H == 0: # H의 배수
    y = H

  print("{}{}".format(y, x.zfill(2)))

간단하게 층부터가 아닌 호실부터 채우는 느낌으로 생각했습니다.

그래서 손님 수에서 호텔층 수의 나머지 값이 방 번호에서 층에 해당합니다.
또, 손님 수에서 호텔층 수의 몫이 방 번호에서 호실 번호가 됩니다.
여기서 그냥 몫을 나눠버리면 소숫점 자리가 나오는데, 무조권 올림을 해줘야 합니다.
내림이 되는 경우가 발생되면 그전의 손님과 방이 겹치게 되기 때문입니다.

if문은 손님 수가 호텔 층 수의 배수가 되어 버리면 값이 0이 되어버리기에 예외처리 했습니다.

마지막으로 출력할때, 층의 자리는 괜찮지만 호실 번호 자리는 1자리 수가 나오면 '01' 느낌으로 무조권 두자리를 만들어줘야 합니다.
그래서, format()을 사용해서 숫자를 붙여서 출력할 수 있게하고, zfill()로 앞의 수를 0으로 채웠습니다.

profile
반갑습니당~😄
post-custom-banner

0개의 댓글