結果
問題 | No.1521 Playing Musical Chairs Alone |
ユーザー |
👑 ![]() |
提出日時 | 2021-05-28 21:01:42 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 843 ms / 2,000 ms |
コード長 | 1,116 bytes |
コンパイル時間 | 350 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 79,100 KB |
最終ジャッジ日時 | 2024-11-07 08:33:17 |
合計ジャッジ時間 | 10,391 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
""""""from sys import stdinimport sysdef matrix_mul(A,B,mod):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]if mod > 0:ans[ai][bj] = now % modelse:ans[ai][bj] = nowreturn ans#行列Aのx乗(当然正方行列じゃないとだめ)def matrix_pow(A,x,mod):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,mod)B = matrix_mul(B,B,mod)x//=2return ansmod = 10**9+7N,K,L = map(int,stdin.readline().split())A = [ [0] * N ]A[0][0] = 1H = [[0] * N for i in range(N)]for i in range(N):for j in range(1,L+1):H[i][(i+j)%N] = 1ans = matrix_mul(A,matrix_pow(H,K,mod),mod)print ("\n".join(map(str,ans[0])))