結果
問題 | No.2857 Div Array |
ユーザー | tassei903 |
提出日時 | 2024-08-25 14:22:37 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 209 ms / 2,000 ms |
コード長 | 1,541 bytes |
コンパイル時間 | 296 ms |
コンパイル使用メモリ | 82,252 KB |
実行使用メモリ | 77,464 KB |
最終ジャッジ日時 | 2024-08-25 14:22:41 |
合計ジャッジ時間 | 3,869 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 36 ms
52,884 KB |
testcase_01 | AC | 36 ms
53,344 KB |
testcase_02 | AC | 42 ms
61,880 KB |
testcase_03 | AC | 44 ms
61,892 KB |
testcase_04 | AC | 41 ms
58,816 KB |
testcase_05 | AC | 41 ms
59,020 KB |
testcase_06 | AC | 37 ms
53,260 KB |
testcase_07 | AC | 44 ms
61,568 KB |
testcase_08 | AC | 44 ms
60,616 KB |
testcase_09 | AC | 45 ms
61,136 KB |
testcase_10 | AC | 47 ms
63,212 KB |
testcase_11 | AC | 182 ms
76,744 KB |
testcase_12 | AC | 73 ms
69,836 KB |
testcase_13 | AC | 51 ms
64,284 KB |
testcase_14 | AC | 148 ms
76,484 KB |
testcase_15 | AC | 55 ms
64,992 KB |
testcase_16 | AC | 43 ms
62,668 KB |
testcase_17 | AC | 133 ms
76,600 KB |
testcase_18 | AC | 128 ms
76,152 KB |
testcase_19 | AC | 86 ms
71,964 KB |
testcase_20 | AC | 180 ms
76,940 KB |
testcase_21 | AC | 209 ms
77,052 KB |
testcase_22 | AC | 181 ms
77,464 KB |
testcase_23 | AC | 182 ms
77,096 KB |
testcase_24 | AC | 47 ms
63,068 KB |
testcase_25 | AC | 46 ms
61,704 KB |
testcase_26 | AC | 46 ms
61,916 KB |
testcase_27 | AC | 46 ms
63,004 KB |
testcase_28 | AC | 36 ms
52,540 KB |
testcase_29 | AC | 37 ms
52,960 KB |
ソースコード
import sys input = lambda :sys.stdin.readline()[:-1] ni = lambda :int(input()) na = lambda :list(map(int,input().split())) yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES") no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO") ####################################################################### def matmul(a, b): n = len(a) m = len(b[0]) l = len(b) c = [[0] * m for _ in range(n)] #print(a, b) for i in range(n): for j in range(m): for k in range(l): c[i][j] += a[i][k] * b[k][j] c[i][j] %= mod return c def matmul2(v, a): n = len(v) m = len(a[0]) c = [0] * m for i in range(n): for j in range(m): c[j] += v[i] * a[i][j] c[j] %= mod return c def matpow(a, n): if n == 0: return [[1 if i == j else 0 for j in range(len(a))] for i in range(len(a))] elif n == 1: return a if n % 2 == 0: return matpow(matmul(a, a), n // 2) return matmul(a, matpow(a, n - 1)) mod = 998244353 n, m, k = na() a = [] for i in range(1, m+1): if not a or a[-1][0] != m // i: a.append((m // i, 1)) else: a[-1] = (m // i, a[-1][1] + 1) #print(len(a)) A = [[0] * len(a) for _ in range(len(a))] for i in range(len(a)): for j in range(len(a)): if abs(a[i][0] - a[j][0]) <= k: A[i][j] = a[j][1] v = [a[i][1] for i in range(len(a))] res = matmul2(v, matpow(A, n-1)) print(sum(res) % mod)