結果
| 問題 |
No.391 CODING WAR
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2020-06-28 17:24:59 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 422 ms / 2,000 ms |
| コード長 | 1,389 bytes |
| コンパイル時間 | 140 ms |
| コンパイル使用メモリ | 82,176 KB |
| 実行使用メモリ | 154,704 KB |
| 最終ジャッジ日時 | 2024-07-07 21:37:03 |
| 合計ジャッジ時間 | 5,229 ms |
|
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 16 |
ソースコード
#yukicoder No.391 CODING WAR
"""
玉:人(区別する)
箱:問題(区別する)
全射
は、包除原理で数え上げる。
問題1に1人以上割り当てる
問題2に1人以上割り当てる...
を余事象にして、
全通り-
問題1に0人割り当てる
問題2に0人割り当てる...
とすれば、あとは0人割り当てるのが何個か何通りかで数え上げれば良い。
"""
import sys
readline = sys.stdin.buffer.readline
def even(n): return 1 if n%2==0 else 0
n,m = map(int,readline().split())
mod = 10**9+7
def pow(n,p,mod=mod): #繰り返し二乗法(nのp乗)
res = 1
while p > 0:
if p % 2 == 0:
n = n ** 2 % mod
p //= 2
else:
res = res * n % mod
p -= 1
return res % mod
def factrial_memo(n=10**6,mod=mod):
fact = [1, 1]
for i in range(2, n + 1):
fact.append((fact[-1] * i) % mod)
return fact
fact = factrial_memo()
def permutation(n,r): #nPr
return fact[n]*pow(fact[n-r],mod-2)%mod
def combination(n,r): #nCr
return permutation(n,r)*pow(fact[r],mod-2)%mod
#return fact[n]*pow(fact[n-r],mod-2)*pow(fact[r],mod-2)
def homogeneous(n,r): #nHr
return combination(n+r-1,r)%mod
#return fact[n+m-1]*pow(fact[n-1],mod-2)*pow(fact[r],mod-2)
ans = 0
for i in range(m):
ans += pow(-1,i)*combination(m,i)*pow(m-i,n)
print(ans%mod)