結果
問題 | No.2857 Div Array |
ユーザー |
![]() |
提出日時 | 2024-08-25 15:45:20 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 857 bytes |
コンパイル時間 | 349 ms |
コンパイル使用メモリ | 82,568 KB |
実行使用メモリ | 102,672 KB |
最終ジャッジ日時 | 2024-08-25 15:45:28 |
合計ジャッジ時間 | 7,010 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 10 TLE * 2 -- * 18 |
ソースコード
from functools import *# 行列 a * b を返すdef mat_mul(a, b) :I, J, K = len(a), len(b[0]), len(b)c = [[0] * J for _ in range(I)]for i in range(I) :for j in range(J) :for k in range(K) :c[i][j] += a[i][k] * b[k][j]c[i][j] %= modreturn c# 行列 x ^ n を返す@lru_cache(maxsize=None)def mat_pow(x, n):y = [[0] * len(x) for _ in range(len(x))]for i in range(len(x)):y[i][i] = 1while n > 0:if n & 1:y = mat_mul(x, y)x = mat_mul(x, x)n >>= 1return ymod = 998244353n, m, k = map(int, input().split())a = tuple([tuple([int(abs(m // (j + 1) - m // (i + 1)) <= k) for j in range(m)]) for i in range(m)])ans = 0for i in mat_mul(mat_pow(a, n - 1), [[1] for _ in range(m)]):ans += i[0]print(ans % mod)