結果
問題 | No.1035 Color Box |
ユーザー |
|
提出日時 | 2021-01-17 10:50:17 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 333 ms / 2,000 ms |
コード長 | 605 bytes |
コンパイル時間 | 87 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 18,432 KB |
最終ジャッジ日時 | 2024-11-29 08:00:19 |
合計ジャッジ時間 | 4,189 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
def prepare(n, MOD):# n! の計算f = 1for m in range(1, n + 1):f *= mf %= MODfn = f# n!^-1 の計算inv = pow(f, MOD - 2, MOD)# n!^-1 - 1!^-1 の計算invs = [1] * (n + 1)invs[n] = invfor m in range(n, 1, -1):inv *= minv %= MODinvs[m - 1] = invreturn fn, invsdef cmb(n,r,fn,invs,mod):return fn*invs[r]*invs[n-r]%modn,m=map(int,input().split())mod=10**9+7fn,invs=prepare(m,mod)a=[0]*(m+1)for i in range(1,m+1):a[i]=(cmb(m,i,fn,invs,mod)*pow(i,n,mod)-a[i-1])%modprint(a[m])