結果
問題 | No.2365 Present of good number |
ユーザー |
![]() |
提出日時 | 2023-07-11 06:48:53 |
言語 | Ruby (3.4.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 940 bytes |
コンパイル時間 | 394 ms |
コンパイル使用メモリ | 7,680 KB |
実行使用メモリ | 12,928 KB |
最終ジャッジ日時 | 2024-09-13 00:01:33 |
合計ジャッジ時間 | 6,318 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 17 WA * 21 RE * 1 |
コンパイルメッセージ
Syntax OK
ソースコード
require 'prime'N, K = gets.split.map(&:to_i)MOD = 10 ** 9 + 7def ff(a, k)return a if k <= 0nums = a.prime_divisionres = 1nums.each do |v, e|res *= (v + 1).pow(e)endff(res, k - 1)enddef f(a, k)return a if k <= 0nums = a.prime_divisionif nums.last[0] <= 3 && nums.size == 2cnt_2 = nums[0][1]cnt_3 = nums[1][1]d = k / 2cnt_2 = (cnt_2 * 2.pow(d, MOD)) % MODcnt_3 = (cnt_3 * 2.pow(d, MOD)) % MODif k % 2 != 0cnt_2, cnt_3 = 2 * cnt_3, cnt_2endreturn (2.pow(cnt_2, MOD) * 3.pow(cnt_3, MOD)) % MODelsif nums.last[0] == 2cnt = nums[0][1]d = k / 2cnt = cnt * 2.pow(d)# cnt = (cnt * 2.pow(d, MOD)) % MODif k % 2 == 0return 2.pow(cnt, MOD)elsereturn 3.pow(2 * cnt, MOD)endendres = 1nums.each do |v, e|res *= (v + 1).pow(e)endf(res, k - 1)end# pp ff(N, K) % MODpp f(N, K) % MOD