結果

問題 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
54,416 KB
testcase_01 AC 47 ms
63,728 KB
testcase_02 AC 47 ms
63,428 KB
testcase_03 AC 51 ms
65,368 KB
testcase_04 AC 46 ms
63,228 KB
testcase_05 AC 45 ms
64,628 KB
testcase_06 AC 46 ms
62,800 KB
testcase_07 AC 48 ms
63,616 KB
testcase_08 AC 57 ms
63,704 KB
testcase_09 AC 53 ms
65,320 KB
testcase_10 AC 62 ms
70,620 KB
testcase_11 AC 360 ms
269,356 KB
testcase_12 AC 118 ms
107,460 KB
testcase_13 AC 76 ms
84,452 KB
testcase_14 AC 380 ms
281,960 KB
testcase_15 AC 95 ms
86,576 KB
testcase_16 AC 48 ms
64,508 KB
testcase_17 AC 325 ms
249,304 KB
testcase_18 AC 260 ms
203,736 KB
testcase_19 AC 174 ms
136,716 KB
testcase_20 AC 430 ms
314,180 KB
testcase_21 AC 412 ms
314,152 KB
testcase_22 AC 413 ms
314,228 KB
testcase_23 AC 416 ms
314,260 KB
testcase_24 AC 428 ms
314,048 KB
testcase_25 AC 405 ms
314,348 KB
testcase_26 AC 439 ms
313,964 KB
testcase_27 AC 408 ms
314,092 KB
testcase_28 AC 42 ms
54,852 KB
testcase_29 AC 39 ms
54,000 KB
権限があれば一括ダウンロードができます

ソースコード

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