結果
問題 |
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 sys input = sys.stdin.readline from 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] %= MOD return C def mat_pow(A, n): res = [[0]*len(A) for _ in range(len(A))] for i in range(len(A)): res[i][i] = 1 while n > 0: if n&1: res = mat_mul(A, res) n >>= 1 A = mat_mul(A, A) return res M, K = map(int, input().split()) MOD = 10**9+7 Y = [[0]*M for _ in range(M)] for i in range(M): for j in range(M): k = (i+j)%M Y[k][i] += 1 k = (i*j)%M Y[k][i] += 1 x = [[0] for _ in range(M)] x[0][0] = 1 res = mat_mul(mat_pow(Y, K), x) print(res[0][0])