結果

問題 No.3243 Multiplication 8 1
ユーザー detteiuu
提出日時 2025-08-22 21:42:42
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 263 ms / 2,000 ms
コード長 780 bytes
コンパイル時間 197 ms
コンパイル使用メモリ 82,180 KB
実行使用メモリ 78,420 KB
最終ジャッジ日時 2025-08-22 21:42:44
合計ジャッジ時間 1,693 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

def matrix(a, b):
    ans = [[0]*len(b[0]) for _ in range(len(a))]
    for i in range(len(a)):
        for j in range(len(b[0])):
            ans[i][j] = sum(a[i][k]*b[k][j]%MOD for k in range(len(b)))%MOD
    return ans

MOD = 998244353

dp = [[[0]*9 for _ in range(9)]]
for i in range(8):
    dp[0][i][i] = 1
    dp[0][i][i^1] = 1
    if 2 <= i:
        dp[0][i][i-2] = 1
        dp[0][i][(i-2)^1] = 1
for i in range(2, 4):
    dp[0][i][6] = 1
dp[0][8][6] = 1
dp[0][8][8] = 1
dp[0][2][8] = 1
dp[0][3][8] = 1
for _ in range(59):
    dp.append(matrix(dp[-1], dp[-1]))

for _ in range(int(input())):
    N = int(input())

    ans = [[0] for _ in range(9)]
    ans[0][0] = 1
    for i in range(60):
        if 1<<i & N:
            ans = matrix(dp[i], ans)
    
    print(ans[6][0])
0