結果
問題 | No.1050 Zero (Maximum) |
ユーザー |
👑 ![]() |
提出日時 | 2020-05-08 22:42:23 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 144 ms / 2,000 ms |
コード長 | 1,298 bytes |
コンパイル時間 | 272 ms |
コンパイル使用メモリ | 82,116 KB |
実行使用メモリ | 76,236 KB |
最終ジャッジ日時 | 2024-07-04 01:02:27 |
合計ジャッジ時間 | 2,221 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 15 |
ソースコード
"""これはダブリングですわどこの要素をいくつ持ってるかでやるzeroとotherさえ分かればいいか行列式の繰り返し二乗法でとけるP+1,12(P-1),2(P-1)のk乗でおk"""mod = 10**9+7#行列A,Bの積(不可能だとエラー)def matrix_mul(A,B):ans = [ [0] * len(B[0]) for i in range(len(A)) ]for ai in range(len(A)):for bj in range(len(B[0])):now = 0for same in range(len(A[0])):now += A[ai][same] * B[same][bj]ans[ai][bj] = now % modreturn ans#行列Aのx乗(当然正方行列じゃないとだめ)def matrix_pow(A,x):B = [[A[i][j] for j in range(len(A[0]))] for i in range(len(A))]ans = [[0] * len(A[0]) for i in range(len(A))]for i in range(len(A)):ans[i][i] = 1while x > 0:if x % 2 == 1:ans = matrix_mul(ans,B)B = matrix_mul(B,B)x//=2return ansM,K = map(int,input().split())fi = [[0] * M]fi[0][0] = 1lis = [ [0] * M for i in range(M) ]for i in range(M):#plusfor j in range(M):lis[i][(i+j) % M] += 1#mulfor j in range(M):lis[i][(i*j) % M] += 1ans = (matrix_mul(fi , matrix_pow(lis,K)))print (ans[0][0] % mod)