[백준] 2292번 (python)

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

 

 

입력

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

출력

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

 

아이디어

해당 번호까지 접근하는데 걸리는 횟수의 경계선의 값들을 보면, 한 번 이동해서 갈 수 있는 최댓값은 7, 그다음은 19, 37, 61.. 이다. 이 번호들의 규칙을 보면 1 + (6*1) , 1+6+6*2 , 1+6+12+6*3 인 것을 알 수 있다. 이를 이용해 i번째 값이 입력해준 N보다 클때까지 반복하는 while문을 이용하여 카운트 수를 늘려주어 이동 횟수를 구한다.

코드

N = int(input())

idx = 1

cnt = 1

while idx < N:

    idx += 6*cnt

    cnt += 1

print(cnt)