AtCoder Beginner Contest 006 B – トリボナッチ数列をPython3で解く

AtCoder上にある問題のうち、AtCoder Problemsでdiff 800以上と判定されているものを順番に解いていく企画。
基本的な考え方は全てコード中のコメントに入れてあるので、参照のこと。

出典：
AtCoder Beginner Contest 006 B – トリボナッチ数列

特に難しいところは無いと思う。おそらく、問題が古いので、今に比べるとdifficultyが高めに出ているんじゃないかな……と思われる。

# AtCoder Beginner Contest 006 B - トリボナッチ数列
# https://atcoder.jp/contests/abc006/tasks/abc006_2
# tag: 基礎問題 数列 MOD

# 特に悩むところの無い問題。変にメモなし再帰で解いたりして
# ハマったりしないように……。

def main():
    n = int(input())
    MOD = 10007

    tribo = [0, 0, 1]

    for _ in range(4, n+1):
        tribo.append((tribo[-1] + tribo[-2] + tribo[-3]) % MOD)

    print(tribo[n-1])

main()

abc006_b.py view raw

関連

コメントを残すコメントをキャンセル