結果
問題 | No.1050 Zero (Maximum) |
ユーザー | neterukun |
提出日時 | 2020-05-08 22:13:08 |
言語 | PyPy3 (7.3.13) |
結果 |
AC
|
実行時間 | 224 ms / 2,000 ms |
コード長 | 910 bytes |
コンパイル時間 | 402 ms |
コンパイル使用メモリ | 87,132 KB |
実行使用メモリ | 77,236 KB |
最終ジャッジ日時 | 2023-09-17 02:58:59 |
合計ジャッジ時間 | 3,403 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge12 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 73 ms
71,388 KB |
testcase_01 | AC | 79 ms
76,388 KB |
testcase_02 | AC | 110 ms
76,644 KB |
testcase_03 | AC | 110 ms
76,976 KB |
testcase_04 | AC | 164 ms
76,508 KB |
testcase_05 | AC | 172 ms
76,708 KB |
testcase_06 | AC | 118 ms
76,724 KB |
testcase_07 | AC | 128 ms
77,104 KB |
testcase_08 | AC | 82 ms
76,284 KB |
testcase_09 | AC | 104 ms
76,744 KB |
testcase_10 | AC | 196 ms
76,692 KB |
testcase_11 | AC | 159 ms
76,724 KB |
testcase_12 | AC | 79 ms
76,712 KB |
testcase_13 | AC | 71 ms
71,352 KB |
testcase_14 | AC | 71 ms
71,468 KB |
testcase_15 | AC | 74 ms
71,424 KB |
testcase_16 | AC | 208 ms
76,676 KB |
testcase_17 | AC | 224 ms
77,236 KB |
ソースコード
def _mul(A, B, MOD): C = [[0] * len(B[0]) for i in range(len(A))] for i in range(len(A)): for k in range(len(B)): for j in range(len(B[0])): C[i][j] = (C[i][j] + A[i][k]*B[k][j]) % MOD return C def pow_matrix(A, n, MOD): """A**nをダブリングによって求める""" B = [[0] * len(A) for i in range(len(A))] for i in range(len(A)): B[i][i] = 1 while n > 0: if n & 1: B = _mul(A, B, MOD) A = _mul(A, A, MOD) n = n // 2 return B m, k = map(int, input().split()) MOD = 10 ** 9 + 7 matrix = [[1] * m for i in range(m)] for i in range(m): for val in range(m): j = i * val % m matrix[i][j] += 1 ans_matrix = pow_matrix(matrix, k, MOD) init = [[0] * m for i in range(m)] init[0][0] = 1 ans = 0 for i in range(m): ans += init[0][i] * ans_matrix[i][0] print(ans % MOD)