結果
問題 | No.1035 Color Box |
ユーザー |
👑 ![]() |
提出日時 | 2020-04-24 21:36:42 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 109 ms / 2,000 ms |
コード長 | 860 bytes |
コンパイル時間 | 150 ms |
コンパイル使用メモリ | 82,340 KB |
実行使用メモリ | 82,048 KB |
最終ジャッジ日時 | 2024-10-15 02:30:44 |
合計ジャッジ時間 | 3,116 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
"""包除原理?m色以下 - mC1*(N-1) + mC2…かな"""def modfac(n, MOD):f = 1factorials = [1]for m in range(1, n + 1):f *= mf %= MODfactorials.append(f)inv = pow(f, MOD - 2, MOD)invs = [1] * (n + 1)invs[n] = invfor m in range(n, 1, -1):inv *= minv %= MODinvs[m - 1] = invreturn factorials, invsdef modnCr(n,r,mod,fac,inv): #上で求めたfacとinvsを引数に入れるべし(上の関数で与えたnが計算できる最大のnになる)return fac[n] * inv[n-r] * inv[r] % modN,M = map(int,input().split())ans = 0mod = 10**9+7fac,inv = modfac(M+10,mod)for i in range(1,M+1):now = pow(i,N,mod) * modnCr(M,i,mod,fac,inv)if i % 2 == M % 2:ans += nowelse:ans -= nowans %= modprint (ans)