結果

問題 No.2857 Div Array
ユーザー shobonvipshobonvip
提出日時 2024-08-25 13:58:46
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 439 ms / 2,000 ms
コード長 750 bytes
コンパイル時間 269 ms
コンパイル使用メモリ 82,460 KB
実行使用メモリ 314,348 KB
最終ジャッジ日時 2024-08-25 13:59:01
合計ジャッジ時間 6,938 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
n, m, k = map(int,input().split())

mat = [[[0] * (m) for i in range(m)] for j in range(30)]

t = defaultdict(int)

for i in range(m):
	t[m//(i+1)] += 1

g = list(t.items())
m = len(g)

mod = 998244353
for i in range(m):
	for j in range(m):
		if abs(g[i][0] - g[j][0]) <= k:
			mat[0][j][i] = g[j][1] % mod

for num in range(29):
	for i in range(m):
		for j in range(m):
			for k in range(m):
				mat[num+1][i][j] += mat[num][i][k] * mat[num][k][j] % mod
			mat[num+1][i][j] %= mod

ans = [g[i][1] for i in range(m)]
n -= 1
for num in range(30):
	if n >> num & 1:
		na = [0] * m
		for i in range(m):
			for j in range(m):
				na[i] += mat[num][i][j] * ans[j] % mod
			na[i] %= mod
		ans = na

print(sum(ans) % mod)
0