2292번 - 벌집

2292번: 벌집

문제

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

입력

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

출력

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

나의 제출 - python

처음에 쉽게 풀려다가 틀려서 많이 수정했다. 벌집의 숫자가 증가하는 패턴의 식을 찾으면 되는 문제이다. 벌집은 한 층이 늘어날때마다 1,(6*1),(6*3),(6*6)···(6*n)으로 늘어난다. 이때 6이 증가하는 순서는 구구단 처럼 1부터 차례로 늘어나는 것이 아닌 점을 유의해야 한다.

# 오답코드
N = int(input())
six = [i*6 for i in range(1,1000000)]
times = 0

for i in six:
    if N > i:
        N = N-i
        times += 1
answer = times + 2
print(answer)

위 코드는 쉽게 풀려고 6을 아예 계산해서 리스트로 만들어버렸다. 결과는 굉장한 오답이라 바로 지우고 새로 코드를 만들었다. 사실 어떤 패턴으로 늘어나는건지 파악하는데는 오래 걸리지 않았지만, 식을 생각하는 게 조금 오래 걸렸다.

N = int(input())
i = 2
times = 1
answer = 0

if N == 1:
    answer = 1

else:
    while True:
        if N <= (6*times)+1:
            break
        times += i
        i += 1
    answer = i
print(answer)

N이 1일 경우 다음 방으로 넘어가지 않기 때문에 answer에 1을 저장한다.

그 외의 경우를 계산할 while문은 무한 반복으로 해 두었다. 6에 곱해주는 숫자를 times로 정의하고, 매 반복마다 times에 i를 더한 후 i를 증가하게 만들었다. times는 1,3,6,10,15··· 로 증가한다. 그리고 N과 비교하는 식은 (6*times)+1 인데, 이 수는 벌집의 한 층의 최대 숫자를 말한다. 이 숫자보다 N이 작으면 N은 그 층에 속한 것이 되므로 break 후 출력한다.