結果

問題 No.2365 Present of good number
コンテスト
ユーザー siman
提出日時 2023-07-11 06:52:44
言語 Ruby
(3.4.1)
結果
WA  
実行時間 -
コード長 936 bytes
コンパイル時間 177 ms
コンパイル使用メモリ 7,680 KB
実行使用メモリ 12,928 KB
最終ジャッジ日時 2024-09-13 00:01:39
合計ジャッジ時間 5,960 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18 WA * 20 RE * 1
権限があれば一括ダウンロードができます
コンパイルメッセージ
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)) % MOD
    cnt_3 = (cnt_3 * 2.pow(d, MOD)) % MOD

    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)
    # cnt = (cnt * 2.pow(d, MOD)) % MOD

    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
0