結果
問題 | No.1521 Playing Musical Chairs Alone |
ユーザー |
|
提出日時 | 2021-04-14 18:31:27 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 423 ms / 2,000 ms |
コード長 | 899 bytes |
コンパイル時間 | 233 ms |
コンパイル使用メモリ | 82,348 KB |
実行使用メモリ | 76,424 KB |
最終ジャッジ日時 | 2024-10-05 03:02:54 |
合計ジャッジ時間 | 5,771 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
#ライブラリ: 行列の計算(乗算/累乗) mod = 10 ** 9 + 7 def mat_mul(a, b): n = len(a) res = [[0] * n for i in range(n)] for i in range(n): for k in range(n): for j in range(n): res[i][j] += a[i][k] * b[k][j] res[i][j] %= mod return res def mat_pow(a, k): if k == 1: return a n = len(a) res = [[0] * n for i in range(n)] for i in range(n): res[i][i] = 1 while k: if k & 1: res = mat_mul(res, a) a = mat_mul(a, a) k >>= 1 return res #入力 n, k, l = map(int, input().split()) #隣接行列の構築 G = [[0] * n for i in range(n)] for i in range(n): for j in range(1, l + 1): G[i][(i + j) % n] = 1 #行列累乗でパスの数を計算 G_ans = mat_pow(G, k) #各 i について答えを出力 for i in range(n): print(G_ans[0][i])