알고리즘&자료구조
[python] 백준9461 - 파도반 수열
Kim_sang_hyeob
2022. 2. 10. 14:58
T=int(input())
P = [1, 1, 1, 2, 2]
for _ in range(T):
N=int(input())
while len(P) <= N :
P.append(P[-1] + P[-5])
print(P[N-1])
정답 코드는 이렇다.
+ )해설
처음에
P.append(P[-1] + P[-5])
여기서 -1 과 -5 대신에 max_len = len(P) 라고 작성하고나서 인덱스를 컨트롤 했는데
다른 사람들의 코드를 보다보니 -1 과 -5 만으로도 간단하게 구할 수 있는 걸 알았다. 문제가 워낙 쉽기도해서 추가적인 설명은 필요 없을 듯 하다.
추가로 숏코딩들을 보다보니까 나는 필요한 구간 까지만구하기 위해서 while 문을 썼는데 그럴 필요 없이 (조건에 주어진 숫자가 매우 작아서 ) 그냥 index 99 까지 모두 구한 뒤 출력한 형태로도 충분히 빠른 듯 하다