[알고리즘 문제] beehive (Python / 파이썬)

문제

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

입력

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

출력

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

---

문제 풀이

각 층의 칸 수 1, 6, 12, 18, ... 이며 

각 층의 최대 값은 7, 19, 37, 61,... 

즉, 칸수를 계속해서 모두 더한 값이다 (1 + 6, 1 + 6 + 12, 1 + 6 + 12 + 18) 

따라서 칸수를 계속 더하다가 만약 주어진 수가 그 범위 안에 있으면

해당 층에 있다는 뜻이므로 층을 출력해주면 된다. 

---

My Code

n = int(input()) # 찾고자 하는 방

num = 1 
i = 1
count = 1
while True:
    if n <= num: # num이 같거나 더 커지면 종료 
        break
    num += (6 * i) # 6의 배수를 계속 더해줌
    count += 1     # 층 수를 세어줌
    i += 1

print(count)