結果

問題 No.2857 Div Array
ユーザー prd_xxxprd_xxx
提出日時 2024-08-25 15:20:34
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 549 bytes
コンパイル時間 183 ms
コンパイル使用メモリ 82,172 KB
実行使用メモリ 114,864 KB
最終ジャッジ日時 2024-08-25 15:20:41
合計ジャッジ時間 6,578 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
60,188 KB
testcase_01 AC 54 ms
67,016 KB
testcase_02 AC 68 ms
75,844 KB
testcase_03 AC 64 ms
76,040 KB
testcase_04 AC 48 ms
67,084 KB
testcase_05 AC 49 ms
68,784 KB
testcase_06 AC 42 ms
63,708 KB
testcase_07 AC 73 ms
76,132 KB
testcase_08 AC 70 ms
76,336 KB
testcase_09 AC 93 ms
76,648 KB
testcase_10 TLE -
testcase_11 TLE -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,M,K = map(int,input().split())
MOD = 998244353

if N==1:
    print(M)
    exit()

mat = [[int(abs(M//(i+1) - M//(j+1)) <= K) for j in range(M)] for i in range(M)]

def mul(l,r):
    ret = [[0]*M for _ in range(M)]
    for i,row in enumerate(l):
        for j,col in enumerate(zip(*r)):
            ret[i][j] = sum(a*b for a,b in zip(row,col)) % MOD
    return ret

X = mat
Z = [[int(i==j) for j in range(M)] for i in range(M)]
N -= 1
while N:
    if N%2:
        Z = mul(Z,X)
    X = mul(X,X)
    N //= 2

print(sum(sum(row)%MOD for row in Z)%MOD)
0