結果

問題 No.2365 Present of good number
ユーザー siman
提出日時 2023-07-11 07:07:46
言語 Ruby
(3.4.1)
結果
AC  
実行時間 107 ms / 2,000 ms
コード長 939 bytes
コンパイル時間 168 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 12,928 KB
最終ジャッジ日時 2024-09-13 00:01:51
合計ジャッジ時間 5,743 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 39
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #
プレゼンテーションモードにする

require 'prime'
N, K = gets.split.map(&:to_i)
MOD = 10 ** 9 + 7
def ff(a, k)
return a if k <= 0
nums = a.prime_division
res = 1
nums.each do |v, e|
res *= (v + 1).pow(e)
end
ff(res, k - 1)
end
def f(a, k)
return a if k <= 0
nums = a.prime_division
if nums.last[0] <= 3 && nums.size == 2
cnt_2 = nums[0][1]
cnt_3 = nums[1][1]
d = k / 2
cnt_2 = (cnt_2 * 2.pow(d, MOD - 1)) % (MOD - 1)
cnt_3 = (cnt_3 * 2.pow(d, MOD - 1)) % (MOD - 1)
if k % 2 != 0
cnt_2, cnt_3 = 2 * cnt_3, cnt_2
end
return (2.pow(cnt_2, MOD) * 3.pow(cnt_3, MOD)) % MOD
elsif nums.last[0] == 2
cnt = nums[0][1]
d = k / 2
cnt = (cnt * 2.pow(d, MOD - 1)) % (MOD - 1)
if k % 2 == 0
return 2.pow(cnt, MOD)
else
return 3.pow(cnt, MOD)
end
end
res = 1
nums.each do |v, e|
res *= (v + 1).pow(e)
end
f(res, k - 1)
end
# pp ff(N, K) % MOD
pp f(N, K) % MOD
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0