結果

問題 No.3310 mod998
コンテスト
ユーザー titia
提出日時 2025-10-27 02:48:17
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 566 ms / 2,000 ms
コード長 727 bytes
コンパイル時間 341 ms
コンパイル使用メモリ 82,656 KB
実行使用メモリ 90,032 KB
最終ジャッジ日時 2025-10-27 02:48:28
合計ジャッジ時間 9,879 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

sys.set_int_max_str_digits(10**6)

mod=998

T=int(input())

for tests in range(T):
    N,M=map(int,input().split())
    K=[int(input()) for i in range(M)]

    if N==0:
        for x in K:
            print(1)
        continue

    X=[1]

    for i in range(1000):
        x=X[-1]*N%mod

        if len(X)>=2 and x == X[1]:
            X.append(x)
            break
        X.append(x)
        
    X=X[1:-1]

    #print(X)
    S=[0]
    for x in X:
        S.append((S[-1]+x)%mod)

    #print(S)

    for k in K:
        if k<len(S):
            print((S[k]+1)%mod)
            continue

        r=k%len(X)
        q=k//len(X)

        ANS=(q*S[-1]+S[r]+1)%mod

        print(ANS)
0