結果

問題 No.2365 Present of good number
ユーザー simansiman
提出日時 2023-07-11 07:07:46
言語 Ruby
(3.3.0)
結果
AC  
実行時間 104 ms / 2,000 ms
コード長 939 bytes
コンパイル時間 72 ms
コンパイル使用メモリ 11,552 KB
実行使用メモリ 15,736 KB
最終ジャッジ日時 2023-10-10 01:35:52
合計ジャッジ時間 5,680 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 98 ms
15,456 KB
testcase_01 AC 96 ms
15,508 KB
testcase_02 AC 98 ms
15,524 KB
testcase_03 AC 96 ms
15,500 KB
testcase_04 AC 98 ms
15,512 KB
testcase_05 AC 96 ms
15,548 KB
testcase_06 AC 98 ms
15,600 KB
testcase_07 AC 97 ms
15,600 KB
testcase_08 AC 97 ms
15,524 KB
testcase_09 AC 98 ms
15,452 KB
testcase_10 AC 98 ms
15,592 KB
testcase_11 AC 99 ms
15,464 KB
testcase_12 AC 100 ms
15,596 KB
testcase_13 AC 97 ms
15,656 KB
testcase_14 AC 98 ms
15,552 KB
testcase_15 AC 98 ms
15,480 KB
testcase_16 AC 98 ms
15,628 KB
testcase_17 AC 98 ms
15,672 KB
testcase_18 AC 98 ms
15,524 KB
testcase_19 AC 99 ms
15,632 KB
testcase_20 AC 99 ms
15,736 KB
testcase_21 AC 100 ms
15,640 KB
testcase_22 AC 97 ms
15,684 KB
testcase_23 AC 99 ms
15,636 KB
testcase_24 AC 98 ms
15,488 KB
testcase_25 AC 97 ms
15,588 KB
testcase_26 AC 100 ms
15,688 KB
testcase_27 AC 97 ms
15,532 KB
testcase_28 AC 101 ms
15,600 KB
testcase_29 AC 96 ms
15,672 KB
testcase_30 AC 98 ms
15,480 KB
testcase_31 AC 97 ms
15,504 KB
testcase_32 AC 97 ms
15,544 KB
testcase_33 AC 97 ms
15,532 KB
testcase_34 AC 99 ms
15,676 KB
testcase_35 AC 104 ms
15,552 KB
testcase_36 AC 102 ms
15,404 KB
testcase_37 AC 101 ms
15,408 KB
testcase_38 AC 97 ms
15,512 KB
testcase_39 AC 101 ms
15,512 KB
testcase_40 AC 98 ms
15,524 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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
0