백준 2292번 파이썬 문제풀이(기본수학 - 벌집)

나의 코드

n = int(input())
a = 1
i = 1
if n == 1:
    print(1)
else:
    while True:
        if n <= a + 6 * i:
            print(i+1)
            break
        a += 6 * i
        i += 1

-> 해당 코드는 급하게 시간내에 푼다고 정신없이 쓴 코드이다. 결국 이문제는 조건만 찾으면 되는데 구간이 6의 배수 만큼 증가한다는 것이다. 그것을 어떻게 코드로 구현할지에 대한 것만 본다는 것이다. 이 코드를 더 간결하게 만들 수 있다.

 

n = int(input())
a = 1
count_six = 1
i = 1

while n > a:
    count_six = 6 * i
    a += count_six
    i += 1

print(i)