백준 2292 벌집 - 파이썬

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

2292번: 벌집

 

 

주어진 숫자가 벌집의 중앙(1)에서부터 몇 번째 줄에 나타나는 지를 찾아야 하는 문제이다.

수학문제이기 때문에 규칙 부터 찾아보았다.

 

숫자(N)	답	규칙(마지막 수와 관련된)
1	1	1 + 6 * 0
2 ~ 7	2	1 + 6 * 0 + 6 * 1
8 ~ 19	3	1 + 6 * 0 + 6 * 1
20 ~ 37	4	1 + 6 * 0 + 6 * 1 + 6 * 2
38 ~ 61	5	1 + 6 * 0 + 6 * 1 + 6 * 2 + 6 * 3
...	...	...

 

첫 번째 줄에는 1개, 두 번째 줄에는 6개, 세 번째 줄에는 12개, 네 번째 줄에는 18개, ... 의 숫자가 배치되는 구조의 벌집이다.

While 문을 사용해서 N에 6 * a (a = 0부터 1씩 증가)를 빼주었고, N이 1과 같거나 더 작아지면 break하는 코드를 짰다.

 

a = 0
ans = 1
N = int(input())

while True:
    N = N - 6*a
    if N <= 1:
        print(ans)
        break
    else:
        a += 1
        ans += 1

 

당연히 거꾸로 1부터 숫자를 더해가면서 풀어도 같은 답이 나온다.

 

ans = 1
n = 1
N = int(input())

while True:
    if n >= N:
        print(ans)
        break
    else:
        n += ans * 6
        ans += 1