結果
問題 | No.1035 Color Box |
ユーザー |
![]() |
提出日時 | 2021-01-26 07:55:35 |
言語 | Ruby (3.4.1) |
結果 |
AC
|
実行時間 | 196 ms / 2,000 ms |
コード長 | 904 bytes |
コンパイル時間 | 39 ms |
コンパイル使用メモリ | 7,680 KB |
実行使用メモリ | 15,616 KB |
最終ジャッジ日時 | 2024-06-23 07:03:49 |
合計ジャッジ時間 | 5,248 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
コンパイルメッセージ
Syntax OK
ソースコード
class ModIntegerattr_reader :fac, :inv, :finv, :modMOD = 10 ** 9 + 7def initialize(n, mod: MOD)@mod = mod@fac = [1, 1]@inv = [1, 1]@finv = [1, 1](2..n).each do |i|@fac[i] = fac[i - 1] * i % mod@inv[i] = mod - inv[mod % i] * (mod / i) % mod@finv[i] = finv[i - 1] * inv[i] % modendenddef combination(n, k)return 0 if n < kreturn 0 if n < 0 || k < 0fac[n] * (finv[k] * finv[n - k] % mod) % modenddef permutation(n, k = n)return 0 if n < kreturn 0 if n < 0 || k < 0fac[n] * (finv[n - k] % mod) % modenddef repeated_combination(n, k)combination(n + k - 1, k)endendN, M = gets.split.map(&:to_i)MOD = 10 ** 9 + 7ans = M.pow(N, MOD)mi = ModInteger.new(M + 1)(M - 1).downto(1) do |x|ans += ((-1) ** (x + M % 2)) * mi.combination(M, x) * x.pow(N, MOD)ans %= MODendputs ans