문제
위의 그림과 같이 육각형으로 이루어진 벌집이 있다. 그림에서 보는 바와 같이 중앙의 방 1부터 시작해서 이웃하는 방에 돌아가면서 1씩 증가하는 번호를 주소로 매길 수 있다. 숫자 N이 주어졌을 때, 벌집의 중앙 1에서 N번 방까지 최소 개수의 방을 지나서 갈 때 몇 개의 방을 지나가는지(시작과 끝을 포함하여)를 계산하는 프로그램을 작성하시오. 예를 들면, 13까지는 3개, 58까지는 5개를 지난다.
입출력
첫째 줄에 N(1 ≤ N ≤ 1,000,000,000)이 주어진다.
입력으로 주어진 방까지 최소 개수의 방을 지나서 갈 때 몇 개의 방을 지나는지 출력한다.
입력 출력 13 3 58 5
N = int(input())
cnt = 1
answer = 1
if N == 1:
print(answer)
else:
for i in range(1, N):
cnt += i*6
if N<=cnt:
answer += i
break
print(answer)
N = int(input())
cnt = 1
i = 1
while N>cnt:
cnt += i*6
i += 1
print(i)
for문이 아니라 while문으로 변경해서 cnt가 N보다 커지면 반복문을 빠져나오게 했다.
모든 케이스를 다 확인해보진 못했지만, 두 코드의 결과가 똑같이 나오는데 왜 첫번째 코드는 틀렸다고 나오는 지 모르겠다.