[Python] 백준 2292 - 벌집

문제

 

위의 그림과 같이 육각형으로 이루어진 벌집이 있다. 그림에서 보는 바와 같이 중앙의 방 1부터 시작해서 이웃하는 방에 돌아가면서 1씩 증가하는 번호를 주소로 매길 수 있다. 숫자 N이 주어졌을 때, 벌집의 중앙 1에서 N번 방까지 최소 개수의 방을 지나서 갈 때 몇 개의 방을 지나가는지(시작과 끝을 포함하여)를 계산하는 프로그램을 작성하시오. 예를 들면, 13까지는 3개, 58까지는 5개를 지난다.

 

입력

첫째 줄에 N(1 ≤ N ≤ 1,000,000,000)이 주어진다.

 

출력

입력으로 주어진 방까지 최소 개수의 방을 지나서 갈 때 몇 개의 방을 지나는지 출력한다.

---

정답

n = int(input())
res = 1
room = 1

while True:
    if n == 1:
        print('1')
        break
    if res >= n and (res - (6 * room)) + 1 <= n:
        print(room)
        break
    else:
        res += 6 * room
        room += 1

위 벌집 그림에 따르면

 

1            ->  1번방 /1

2 ~ 7      -> 2번방 /6

8 ~ 19    -> 3번방 /12

20 ~ 37 -> 4번방 /18

38 ~ 61  -> 5번방 / 24

 

이런식으로 방을 거쳐야하는 번호들이 6의배수로 증가하는걸 볼 수 있다. 

 

만약 n=1이라면 1을 출력해준다.

 

만약 n이 1이 아닐 경우에는 res가 n보다 작고, (res - 6*room) + 1이 성립 될 경우에 room(방 번호)를 출력해준다.

성립이 되지 않는다면 res에 6*room을 더해주고, room을 1 증가시켜준다.

여기서 res는 7, 19, 37, 61처럼 방의 마지막 숫자이다.

 

예를 들어 n = 21일 경우, 

n은 37보다 작고, (37 - 6*3) +1 보다 크기 때문에 4가 출력이 된다.