[BOJ] 백준 2292 벌집 python

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

2292번: 벌집

그림이 있어 친절하다

우선 끝 숫자를 주목했다. 그보다 더 크면 한바퀴 더 도는 식으로 그림이 그려졌다.

1 -> 7 -> 19 -> 37 -> 61 ....

여기에 무슨 규칙이 있는 걸까?

고등학교 때 배웠던 계차함수 따위가 떠오른다. (기억 안남)

항과 항 사이의 차이를 관찰해보면 6 -> 12 -> 18 -> 24 ... 씩 더해지고 있는 것을 알 수 있다.

규칙을 발견했으면 코드로 옮기는 것은 더욱 수월하다. 

6 * (1 + 2 + 3 + 4 + ...) 이므로 고딩때 배웠던 합 공식에 따라 6 * n * (n + 1) // 2 로 식을 만들 수 있다.

이 식이 떠오르지 않아 만들 수 없다면 반복문 돌면서 더하면 되니까 걱정 안해도 된다.

기억만큼이나 흐릿한 화질

 

코드는 아래와 같다.

number = int(input())

def solution(number):
	n = 1
	while True:
		if number == 1:
			return 1
		if number <= 3 * n * (n + 1) + 1:
			return n + 1
		n += 1

print(solution(number))