[COS Pro 1급, Python] 5차 1번 : 우리는 계단도 특별하게 오르죠

문제 유형

 빈칸

난이도

 easy

Note 

 1. 수열 문제. 1 2 4 7 13 24 . . .

 

Code

# -*- coding: utf-8 -*-
# UTF-8 encoding when using korean
def solution(n):
	answer = 0
	steps = [0 for _ in range(n+1)]
	steps[1] = 1
	steps[2] = 2
	steps[3] = 4
	for i in range(4, n+1):
		steps[i] = steps[i-1] + steps[i-2] + steps[i-3]
	print(steps)
	answer = steps[n]
	return answer

n1 = 3
ret1 = solution(n1)

print("solution 함수의 반환 값은", ret1, "입니다.")

n2 = 4
ret2 = solution(n2)

print("solution 함수의 반환 값은", ret2, "입니다.")

 

※ 가끔 코드 중 print(~)가 있습니다. 정리 못한 점 죄송합니다.