BOJ/Python

백준 9461번 파도반 수열 파이썬

띵지니어 2022. 9. 19. 14:55

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

 

9461번: 파도반 수열

오른쪽 그림과 같이 삼각형이 나선 모양으로 놓여져 있다. 첫 삼각형은 정삼각형으로 변의 길이는 1이다. 그 다음에는 다음과 같은 과정으로 정삼각형을 계속 추가한다. 나선에서 가장 긴 변의

www.acmicpc.net

 

내 답안

import sys
input = sys.stdin.readline

dp = [0] * 101
dp[0] = dp[1] = dp[2] = 1

for i in range(int(input())):
    n = int(input())
    if n < 3:
        print(dp[n])
    else:
        for i in range(n-2):
            dp[i+3] = dp[i] + dp[i+1]
        print(dp[n-1])

REVIEW

 

수열에 규칙을 찾고, ( a3 = a0+a1 )

그냥 피보나치 수열 만드는 것 처럼 dp배열을 만들고 코드를 짰다.

dp[i+3] = dp[i] + dp[i+1]