백준 1568번 새

 Description

N마리의 새가 나무에 앉아있고, 자연수를 배우기 원한다. 새들은 1부터 모든 자연수를 오름차순으로 노래한다. 어떤 숫자 K를 노래할 때, K마리의 새가 나무에서 하늘을 향해 날아간다. 만약, 현재 나무에 앉아있는 새의 수가 지금 불러야 하는 수 보다 작을 때는, 1부터 게임을 다시 시작한다.

나무에 앉아 있는 새의 수 N이 주어질 때, 하나의 수를 노래하는데 1초가 걸린다고 하면, 모든 새가 날아가기까지 총 몇 초가 걸리는지 출력하는 프로그램을 작성하시오.

 Input

첫째 줄에 새의 수 N이 주어진다. 이 값은 109보다 작거나 같다.

 Output

첫째 줄에 정답을 출력한다.

 Solution

# 1차 시도

n = int(input())
arr = []

for x in range(n):
    arr.append(x+1)

answer = 0
while(len(arr)>0):
    count = 1
    while len(arr) >= count:
        for _ in range(count):
            arr.pop(0)
        count+=1
        answer+=1

print(answer)
  • 메모리 초과 판정을 받았다.
# 2차 시도

n = int(input())

answer = 0

while(n>0):
    count = 1
    while n >= count:
        for _ in range(count):
            n-=1
        count+=1
        answer+=1

print(answer)
  • 3중 반복문으로 시간 초과 위험이 있었으나 PyPy3으로 했더니 통과되었다.

 Feedback

# 다른 사람 풀이

n = int(input())
result = 0
k = 1

while n!=0:
    if k > n:
        k = 1
    n -= k
    k +=1
    result +=1

print(result)

Source