結果
問題 | 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 + 7def 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] %= modreturn resdef mat_pow(a, k):if k == 1: return an = len(a)res = [[0] * n for i in range(n)]for i in range(n):res[i][i] = 1while k:if k & 1:res = mat_mul(res, a)a = mat_mul(a, a)k >>= 1return 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])