백준 문제 중 2292번, 구현, ‘벌집’

백준 문제 중 2292번

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

문제

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

입력

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

출력

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

---

풀이

주어진 문제의 규칙과 조건을 찾아보면 한겹이 늘어날때마다 추가 되는 수의 개수가 6개씩 증가하고 총 육각형의 개수는 차례대로 방의 수로 매겨지고 방의 수는 아래와 같다.

• 1단계
  • 1개
• 2단계
  • (1 + 1*6) 개
• 3단계
  • (1 + 16 + 26) 개
• 4단계
  • (1 + 16 + 26 + 3*6) 개

…

이를 코드로 구현하면

def comb(n:int) ->int:

    total = 1
    i = 0
  
    while True:
        total = total + i*6
        i+=1
        if n<=total:
            return i

num = int(input())

comb(num)

37





4

배운점

문제의 조건에 맞게 규칙을 잘 관찰하자