結果
問題 | No.1521 Playing Musical Chairs Alone |
ユーザー | shotoyoo |
提出日時 | 2021-05-28 21:22:04 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 60 ms / 2,000 ms |
コード長 | 1,050 bytes |
コンパイル時間 | 859 ms |
コンパイル使用メモリ | 82,116 KB |
実行使用メモリ | 62,208 KB |
最終ジャッジ日時 | 2024-11-07 08:55:49 |
合計ジャッジ時間 | 3,313 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
import sys input = lambda : sys.stdin.readline().rstrip() sys.setrecursionlimit(2*10**5+10) write = lambda x: sys.stdout.write(x+"\n") debug = lambda x: sys.stderr.write(x+"\n") writef = lambda x: print("{:.12f}".format(x)) n,k,l = list(map(int, input().split())) M = 10**9+7 def pmul(a,b): n = len(a) m = len(b) c = [0]*(n+m-1) for i in range(n): for j in range(m): c[i+j] += a[i]*b[j]%M if c[i+j] > M: c[i+j] -= M return c def pmulk(a,k): ans = [1] while k: if k&1: ans = pmul(ans, a) nans = [0]*n for i in range(len(ans)): nans[i%n] += ans[i] nans[i%n] %= M ans = nans k = k>>1 a = pmul(a,a) na = [0]*n for i in range(len(a)): na[i%n] += a[i] na[i%n] %= M a = na # print(len(ans)) return ans[:n] if n==l: a = [1]*n else: a = [0] + [1]*l + [0]*(n-l-1) ans = pmulk(a, k) write("\n".join(map(str, ans)))