結果

問題 No.2365 Present of good number
ユーザー a01sa01toa01sa01to
提出日時 2023-06-30 22:31:51
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,370 bytes
コンパイル時間 2,135 ms
コンパイル使用メモリ 222,028 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-07 10:23:02
合計ジャッジ時間 3,424 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 AC 1 ms
6,940 KB
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 AC 1 ms
6,940 KB
testcase_18 AC 1 ms
6,944 KB
testcase_19 WA -
testcase_20 AC 2 ms
6,944 KB
testcase_21 AC 1 ms
6,940 KB
testcase_22 WA -
testcase_23 AC 2 ms
6,944 KB
testcase_24 AC 2 ms
6,940 KB
testcase_25 WA -
testcase_26 AC 2 ms
6,944 KB
testcase_27 AC 1 ms
6,944 KB
testcase_28 AC 2 ms
6,944 KB
testcase_29 AC 1 ms
6,944 KB
testcase_30 AC 2 ms
6,944 KB
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 AC 1 ms
6,944 KB
testcase_37 AC 2 ms
6,940 KB
testcase_38 AC 2 ms
6,944 KB
testcase_39 AC 2 ms
6,944 KB
testcase_40 AC 2 ms
6,940 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;
void operator<<(ostream& os, const mint& m) { os << m.val(); }

int solve(int n, ll k) {
  Debug(n, k);
  vector<pair<ll, mint>> 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 (!pfcond) break;

    --k;
    map<ll, mint> 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;
  for (auto [p, e] : pf) {
    if (p == 2) {
      ans *= mint(k % 2 == 1 ? 3 : 2).pow((e * mint(2).pow(k / 2)).val());
    }
    else if (p == 3) {
      ans *= mint(k % 2 == 0 ? 3 : 4).pow((e * mint(2).pow(k / 2)).val());
    }
    else {
      ans *= mint(p).pow(e.val());
    }
  }
  return ans.val();
}

int solve_greedy(ll n, ll k) {
  if (k == 0) return n;
  vector<pair<int, int>> 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 });
  ll res = 1;
  for (auto [p, e] : pf) {
    rep(_, e) res *= p + 1;
  }
  return solve_greedy(res, k - 1);
}

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