프로그래밍/파이썬 공부

[백준 #2292][Python] 벌집

winblack22 2022. 1. 12. 22:29

문제

#2292 벌집

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

 

풀이

 

#2292 문제 풀이

벌집을 들여다보면 규칙이 보인다.

육각형인 벌집의 모양에 따라 1부터 시작해서 6의 배수씩 누적되어 방의 개수(cnt)를 셀 수 있다.

여기서 주목해야 할건 6의 배수씩 "누적" 된다는 것인데

처음엔 누적은 모르고 그냥 6으로 나누고 몫과 나머지를 고려하여 코드를 짰더니 13까지 2개의 방만 지나간다는 공간을 초월한 답이 나왔다.

누적을 고려하며 처음 입력받은 수 보다 커지지 않는 선에서 방의 개수(cnt)를 추가하면 되겠다 생각이 들었다.

n = int(input())

cnt = 1
num = 1

while n > num:
    num += cnt * 6
    cnt += 1

print(cnt)