結果
問題 |
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)