結果

問題 No.1259 スイッチ
ユーザー simansiman
提出日時 2023-06-29 11:44:27
言語 Ruby
(3.4.1)
結果
AC  
実行時間 308 ms / 2,000 ms
コード長 580 bytes
コンパイル時間 355 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 20,096 KB
最終ジャッジ日時 2024-07-06 03:06:55
合計ジャッジ時間 14,226 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 61
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

class Integer
  def mod_inverse(mod)
    self.pow(mod - 2, mod)
  end
end

N, K, M = gets.split.map(&:to_i)
MOD = 10 ** 9 + 7
ans = 0
memo = Array.new(N + 1, 0)
memo[0] = 1
val = 1
1.upto(N - 1) do |l|
  val *= (N - l)
  val %= MOD
  memo[l] = val
end

if M == 1
  1.upto(N) do |ls|
    next if K % ls != 0

    ans += memo[ls - 1] * N.pow(N - ls, MOD)
    ans %= MOD
  end
else
  ans = N.pow(N, MOD)

  1.upto(N) do |ls|
    next if K % ls != 0

    ans += -1 * memo[ls - 1] * N.pow(N - ls, MOD)
    ans %= MOD
  end

  ans *= (N - 1).mod_inverse(MOD)
  ans %= MOD
end

puts ans
0