結果
問題 | No.1050 Zero (Maximum) |
ユーザー |
![]() |
提出日時 | 2020-05-08 23:10:20 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 122 ms / 2,000 ms |
コード長 | 882 bytes |
コンパイル時間 | 136 ms |
コンパイル使用メモリ | 82,072 KB |
実行使用メモリ | 76,712 KB |
最終ジャッジ日時 | 2024-07-04 01:10:20 |
合計ジャッジ時間 | 2,094 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 15 |
ソースコード
import sysinput = sys.stdin.readlinefrom collections import *def mat_mul(A, B):C = [[0]*len(B[0]) for _ in range(len(A))]for i in range(len(A)):for j in range(len(B[0])):for k in range(len(A[0])):C[i][j] += A[i][k]*B[k][j]C[i][j] %= MODreturn Cdef mat_pow(A, n):res = [[0]*len(A) for _ in range(len(A))]for i in range(len(A)):res[i][i] = 1while n > 0:if n&1:res = mat_mul(A, res)n >>= 1A = mat_mul(A, A)return resM, K = map(int, input().split())MOD = 10**9+7Y = [[0]*M for _ in range(M)]for i in range(M):for j in range(M):k = (i+j)%MY[k][i] += 1k = (i*j)%MY[k][i] += 1x = [[0] for _ in range(M)]x[0][0] = 1res = mat_mul(mat_pow(Y, K), x)print(res[0][0])