結果

問題 No.658 テトラナッチ数列 Hard
コンテスト
ユーザー kichirb3
提出日時 2018-03-06 18:34:38
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 53 ms / 2,000 ms
コード長 1,069 bytes
コンパイル時間 75 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,776 KB
最終ジャッジ日時 2024-09-19 19:57:54
合計ジャッジ時間 1,094 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

# -*- coding: utf-8 -*-
"""
No.658 テトラナッチ数列 Hard
https://yukicoder.me/problems/no/658

"""
import sys
from sys import stdin
input = stdin.readline


def solve(limit):
    loop = 0
    loop_check= dict()
    res = [None, 0, 0, 0, 1]
    for i in range(5, limit+1):
        h = '{:02d}{:02d}{:02d}{:02d}'.format(res[-4], res[-3], res[-2], res[-1])
        if h in loop_check:
            loop = loop_check[h]
            break
        else:
            loop_check[h] = i
        tmp = sum(res[-4:]) % 17
        res.append(tmp)
    return res, loop


def main(args):
    Q = int(input())
    queries = [int(input()) for _ in range(Q)]

    tetra, looped = solve(max(queries))
    if looped != 0:
        loop = len(tetra) - looped
        for q in queries:
            if q % loop == 0:
                print(tetra[looped])
            elif q > loop:
                print(tetra[q % loop])
            else:
                print(tetra[q])
    else:
        for q in queries:
            print(tetra[q])


if __name__ == '__main__':
    main(sys.argv[1:])
0