[백준] Bronze 2. 2292 - 벌집 python

문제 링크

https://www.acmicpc.net/problem/2292

---

문제 설명

 

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

 

입력

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

 

출력

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

---

제출 코드

import sys

input = sys.stdin.readline

n = int(input().rstrip())
cnt = 1
i_max = 1

while True:
    if i_max < n:
        i_max += 6*cnt
        cnt += 1
    if i_max >= n:
        print(cnt)
        break

 

풀이

• 1에서 N 까지 도달하려면 몇 겹의 벌집방을 지나야 하는지 계산해야 한다.
• 각 겹에서 최댓값은 1 + 6 * n(겹수-1) 이므로, 입력받은 N이 각 겹의 최댓값보다 크다면 다음 겹으로 넘어가서 count 한다.

알아둘 사항

• 벌집의 규칙을 생각하는게 생각보다 어려웠다. 좀 더 쉽게 접근하도록 해보자.