파이썬) 벌집 - BOJ

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

2292번: 벌집

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

 

풀이

문제는 다음과 같다

n이 1이라면 1칸, 2라면 2칸, 5라면 2칸, 90이라면 6칸이다.

처음 이 문제를 접했을땐 dp로 풀어보려 했으나 조금만 생각하니 dp를 사용하지 않아도 되는 규칙을 찾아냈다.

1은 한칸

2부터 7은 2칸,

8부터 19는 3칸

20부터 37은 4칸

38부터 61은 5칸

62부터 91은 6칸이다.

매 맥시멈에서 다음 맥시멈으로 넘어가기까지는 i(1부터 1씩 증가하는 숫자) 곱하기 6이라는 숫자가 필요하다

즉, 2번째 칸이라면 i = 1, i * 6 = 7

만약 3번째 칸이라면, i = 2, i * 6 = 12, 12 + 이번 맥시멈(7) = 19...

이를 코드로 옮긴다면 다음과 같다.

n = int(input())

if n == 1:
    print(1)
else:

    max_n = 1000000000

    loop = max_n // 6

    # dp = [7]
    cnt = 1
    m = 2
    for i in range(1, n):
        cnt += (6 * i)

        if n <= cnt:
            print(m)
            break
        else:
            m += 1