結果

問題 No.2600 Avator Height
ユーザー lam6er
提出日時 2025-03-20 21:04:49
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 138 ms / 2,000 ms
コード長 821 bytes
コンパイル時間 168 ms
コンパイル使用メモリ 82,436 KB
実行使用メモリ 121,392 KB
最終ジャッジ日時 2025-03-20 21:05:02
合計ジャッジ時間 6,601 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

mod = 998244353
max_n = 200000

def main():
    input = sys.stdin.read().split()
    Q = int(input[0])
    ns = list(map(int, input[1:Q+1]))
    
    # Precompute R sequence
    r = [0] * (max_n + 1)
    r[1] = 1
    if max_n >= 2:
        r[2] = 1
    for i in range(3, max_n + 1):
        r[i] = (r[i-1] + r[i-2]) % mod
    
    # Precompute E sequence
    e = [0] * (max_n + 1)
    e[1] = 1
    if max_n >= 2:
        e[2] = 3
    for i in range(3, max_n + 1):
        e[i] = (e[i-1] + e[i-2]) % mod
    
    # Precompute answers
    ans = [0] * (max_n + 1)
    for i in range(1, max_n + 1):
        rsq = (r[i] * r[i]) % mod
        esq = (e[i] * e[i]) % mod
        ans[i] = (5 * rsq - esq) % mod
    
    # Process each query
    for n in ns:
        print(ans[n])

if __name__ == '__main__':
    main()
0