타일 장식 / python

문제설명

대구 달성공원에 놀러 온 지수는 최근에 새로 만든 타일 장식물을 보게 되었다. 타일 장식물은 정사각형 타일을 붙여 만든 형태였는데, 한 변이 1인 정사각형 타일부터 시작하여 마치 앵무조개의 나선 모양처럼 점점 큰 타일을 붙인 형태였다. 타일 장식물의 일부를 그리면 다음과 같다.

그림에서 타일에 적힌 수는 각 타일의 한 변의 길이를 나타낸다. 타일 장식물을 구성하는 정사각형 타일 한 변의 길이를 안쪽 타일부터 시작하여 차례로 적으면 다음과 같다. [1, 1, 2, 3, 5, 8, …] 지수는 문득 이러한 타일들로 구성되는 큰 직사각형의 둘레가 궁금해졌다. 예를 들어, 처음 다섯 개의 타일이 구성하는 직사각형(위에서 빨간색으로 표시한 직사각형)의 둘레는 26이다.

타일의 개수 N이 주어질 때, N개의 타일로 구성된 직사각형의 둘레를 return 하도록 solution 함수를 작성하시오.

제한 사항

N은 1 이상 80 이하인 자연수이다.

입출력 예

5	26
6	42

---

풀이

이 문제는 동적프로그래밍으로 차근차근 작은 거 부터 실행해나가며 된다. 

def solution(N):
    #메모이제이션
    dp = [0] * int(N+2)
    
    if N == 1:
        return 4
    if N == 2:
        return 6
    dp[0], dp[1], dp[2] = 0, 1, 1
    
    for i in range(3, N+2):
        dp[i] = dp[i-1] + dp[i-2]

    return 2*dp[N-1] + 4*dp[N]