[BOJ] 2292. 벌집(python)

📌문제

💪아이디어

1. 집합 묶기
   1에서 출발하여 입력값에 도달하기 위해 방문하는 최소의 방의 개수를 찾기 위해서는 입력값이 몇 번째 그룹에 속하는지 알아야한다.
   [1], [2,3,4,5,6,7],[8,9,10,..,19],[20,...,37],...
   으로 집합이 묶인다. 각 집합의 첫 번째 수는 1,2,8,20,...으로 1,6,12,18,...씩 늘어난다. 전 첫번째 숫자+6(n-2) 의 규칙성이 있다.
   

2. 입력값이 속하는 집합의 인덱스 찾기
   입력값이 다음 집합의 첫 번째보다 작을 때 집합을 찾을 수 있다. 따라서 위의 점화식을 사용하여 입력값보다 큰 값을 찾아 집합의 인덱스 -1한다.

🥂코드

n=int(input())
arr=[0,1,2]
if n==1: print(1)
else:
    answer=3
    while arr[-1]<=n:
        arr.append(arr[-1] + 6*(answer-2))
        answer+=1
    print(answer-2)