結果

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

ソースコード

diff # 

# -*- 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_start = -1                    #  res[]中の数字の並びで繰り返しが発生するスタート地点
    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_start = loop_check[h]
            break
        else:
            loop_check[h] = i
        tmp = sum(res[-4:]) % 17
        res.append(tmp)
    return res, loop_start


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

    tetra, loop_start = solve(max(queries))
    tetra_mod = tetra[loop_start:] #  繰り返し部分だけのテーブル

    orikaeshi = len(tetra)          #  この数字以上は繰り返しパタンに入っている
    cycle =  orikaeshi - loop_start #  繰り返しの周期
    for q in queries:
        if q >= orikaeshi:
            print(tetra_mod[(q-loop_start)%cycle])
            # print(tetra[(q-loop_start)%cycle + loop_start])
        else:
            print(tetra[q])


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