結果

問題 No.1839 Concatenation Matrix
ユーザー gew1fw
提出日時 2025-06-12 16:21:00
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 776 bytes
コンパイル時間 191 ms
コンパイル使用メモリ 82,276 KB
実行使用メモリ 268,528 KB
最終ジャッジ日時 2025-06-12 16:21:21
合計ジャッジ時間 6,242 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 7 TLE * 1 -- * 8
権限があれば一括ダウンロードができます

ソースコード

diff #

MOD = 998244353
MOD_PHI = MOD - 1  # Since MOD is prime

def main():
    import sys
    input = sys.stdin.read().split()
    N = int(input[0])
    a = list(map(int, input[1:N+1]))
    
    # Precompute p_i = 10^(2^{i-2} mod MOD_PHI) mod MOD for i >= 2
    p = [0] * (N + 1)  # p[2] is for i=2, etc.
    if N >= 2:
        p[2] = 10 % MOD
    for i in range(3, N + 1):
        exponent = pow(2, i - 2, MOD_PHI)
        p[i] = pow(10, exponent, MOD)
    
    # Initialize val array
    val = a.copy()
    for i in range(2, N + 1):
        new_val = [0] * N
        for j in range(N):
            next_j = (j + 1) % N
            new_val[j] = (val[j] * p[i] + val[next_j]) % MOD
        val = new_val
    
    for x in val:
        print(x)

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