본문 바로가기
Beakjoon&프로그래머스/파이썬

[백준/파이썬] 15989번 1, 2, 3 더하기 4

by 현장 2021. 12. 27.

-코드

for _ in range(int(input())):
    n = int(input())
    dp = [1, 1, 2, 3]
    for i in range(4, n + 1):
        num = dp[-1] + dp[-2] - dp[-3]
        if i % 3 == 0:
            num += 1
        dp.append(num)
    print(dp[n])

처음에 수식을 잘못 짜서 틀린 입출력이 나와서 다시 규칙을 고민하고 찾아봐서 해결했으나 특정수에서 값이 달라져서 6 이상부터 문제가 생겨서 보니 1이 낮게 입력이 된 부분을 찾아서 해결해 주었습니다.