結果
問題 | No.2857 Div Array |
ユーザー |
![]() |
提出日時 | 2024-09-03 20:06:09 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,023 bytes |
コンパイル時間 | 631 ms |
コンパイル使用メモリ | 82,328 KB |
実行使用メモリ | 76,520 KB |
最終ジャッジ日時 | 2024-09-03 20:06:14 |
合計ジャッジ時間 | 4,407 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 21 WA * 9 |
ソースコード
mod = 998244353def matrix_prod(A,B):l,m,n = len(A),len(B),len(B[0])C = [[0 for _ in range(n)] for _ in range(l)]for i in range(l):for j in range(m):a = A[i][j]for k in range(n):C[i][k] = (C[i][k] + a * B[j][k] % mod) % modreturn Cdef matrix_pow(A,n):N = len(A)if n % 2:return matrix_prod(A,matrix_pow(A,n-1))elif n > 0:B = matrix_pow(A,n//2)return matrix_prod(B,B)else:res = [[int(i==j) for j in range(N)] for i in range(N)]return resN,M,K = map(int,input().split())C = [0 for _ in range(M + 1)]for i in range(1,M + 1):C[M // i] += 1X = []for i in range(1,M + 1):if i ** 2 < M:X.append(M // i)if i ** 2 <= M:X.append(i)X.sort()n = len(X)A = [[0 for _ in range(n)] for _ in range(n)]for i in range(n):for j in range(n):if abs(X[i] - X[j]) <= K:A[i][j] = C[X[i]]dp = [[C[X[i]]] for i in range(n)]dp = matrix_prod(matrix_pow(A,N - 1),dp)ans = 0for i in range(n):ans += dp[i][0]print(ans % mod)