알고리즘/백준

[백준] 9461번 파도반 수열 (파이썬)

알감자 2022. 1. 1. 23:28
 

9461번: 파도반 수열

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

www.acmicpc.net

 

코드

import sys

n = int(input())
li = list(sys.stdin.readline().rsplit() for i in range(n))

def count(l):
    base = [1,1,1,2,2,3]
    answer = 0
    for i in range(5,l-1):
        answer = base[i] + base[i-4]
        base.append(answer)
    print(base[l-1])

for i in li:
    count(int(i[0]))