[baekjoon]python #14606 피자 (Small)

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

14606번: 피자 (Small)

🎈각 box의 층 수에 따라 최댓값을 계산해보니 다음과 같은 규칙을 발견할 수 있었다.

🎈이 규칙을 그대로 코드에 적용시켜보았다.

n = int(input())
arr = [0] #index와 box의 층수를 맞추기 위해 0번 index에 0을 넣어 arr 초기화
def pizza(box):
    if box == n: #내가 구하고자 하는 n까지 배열 채우고 리턴
        arr.append(arr[box-1]+box-1)
        return

    arr.append(arr[box-1]+box-1)
    pizza(box+1)

pizza(1) #1층부터 시작!
print(arr[n]) #원하는 층의 최대값 출력