結果

問題 No.2365 Present of good number
ユーザー a01sa01toa01sa01to
提出日時 2023-06-30 23:22:40
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 2,261 bytes
コンパイル時間 2,031 ms
コンパイル使用メモリ 217,852 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-07 11:15:57
合計ジャッジ時間 3,015 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 1 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 1 ms
5,376 KB
testcase_10 AC 1 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 1 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 1 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 1 ms
5,376 KB
testcase_18 AC 1 ms
5,376 KB
testcase_19 AC 1 ms
5,376 KB
testcase_20 AC 1 ms
5,376 KB
testcase_21 AC 1 ms
5,376 KB
testcase_22 AC 1 ms
5,376 KB
testcase_23 AC 1 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 1 ms
5,376 KB
testcase_28 AC 1 ms
5,376 KB
testcase_29 AC 1 ms
5,376 KB
testcase_30 AC 1 ms
5,376 KB
testcase_31 AC 2 ms
5,376 KB
testcase_32 AC 2 ms
5,376 KB
testcase_33 AC 1 ms
5,376 KB
testcase_34 AC 2 ms
5,376 KB
testcase_35 AC 1 ms
5,376 KB
testcase_36 AC 1 ms
5,376 KB
testcase_37 AC 1 ms
5,376 KB
testcase_38 AC 2 ms
5,376 KB
testcase_39 AC 1 ms
5,376 KB
testcase_40 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#ifdef LOCAL
  #include "settings/debug.cpp"
  #define _GLIBCXX_DEBUG
#else
  #define Debug(...) void(0)
#endif
using ll = long long;
#define rep(i, n) for (int i = 0; i < (n); ++i)

#include <atcoder/modint>
using mint = atcoder::modint1000000007;
using mint2 = atcoder::static_modint<1000000006>;
void operator<<(ostream& os, const mint& m) { os << m.val(); }

int solve(int n, ll k) {
  Debug(n, k);
  vector<pair<ll, ll>> pf(0);
  for (int i = 2; i * i <= n; ++i) {
    if (n % i == 0) {
      pf.push_back({ i, 0 });
      while (n % i == 0) {
        n /= i;
        pf.back().second++;
      }
    }
  }
  if (n > 1) pf.push_back({ n, 1 });
  Debug(pf);
  while (k > 0) {
    bool pfcond = false;
    if (pf.size() > 2) pfcond = true;
    for (auto [p, e] : pf)
      if (p != 2 && p != 3) pfcond = true;
    if (k % 2 == 1) pfcond = true;
    if (!pfcond) break;

    --k;
    map<ll, ll> new_pf;
    for (auto [p, e] : pf) {
      ll new_p = p + 1;
      for (int i = 2; i * i <= new_p; i++) {
        while (new_p % i == 0) {
          new_p /= i;
          new_pf[i] += e;
        }
      }
      if (new_p != 1) {
        new_pf[new_p] += e;
      }
    }
    pf.clear();
    for (auto [p, e] : new_pf) {
      pf.push_back({ p, e });
    }
    Debug(k, pf);
  }
  mint ans = 1;
  if (k == 0) {
    for (auto [p, e] : pf) {
      ans *= mint(p).pow(e);
    }
  }
  else {
    assert(k % 2 == 0);
    for (auto [p, e] : pf) {
      if (p == 2) {
        ans *= mint(2).pow(mint2(e * mint2(2).pow(k / 2).val()).val());
      }
      else if (p == 3) {
        ans *= mint(3).pow(mint2(e * mint2(2).pow(k / 2).val()).val());
      }
      else {
        assert(false);
      }
    }
  }
  return ans.val();
}

int main() {
  ll n, k;
  cin >> n >> k;
  // cout << ans.val() << endl;
  // for (int n = 2; n < 10000; n++) {
  //   for (int k = 1; k < 50000; k++) {
  //     int ans = solve(n, k);
  //     int ans_greedy = solve_greedy(n, k);
  //     if (ans_greedy == -1) {
  //       Debug(n, k, "OVERFLOW");
  //       break;
  //     }
  //     if (ans != ans_greedy) {
  //       Debug(n, k, ans, ans_greedy);
  //       return 0;
  //     }
  //   }
  // }
  cout << solve(n, k) << endl;
  return 0;
}
0