結果

問題 No.2365 Present of good number
コンテスト
ユーザー ___Johniel
提出日時 2023-06-30 22:51:34
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 4,263 bytes
コンパイル時間 2,269 ms
コンパイル使用メモリ 204,904 KB
最終ジャッジ日時 2025-02-15 04:26:12
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 6 WA * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

// github.com/Johniel/contests
// yukicoder/2365/main.cpp

#include <bits/stdc++.h>

#define each(i, c) for (auto& i : c)
#define unless(cond) if (!(cond))
// #define endl "\n"

using namespace std;

template<typename P, typename Q> ostream& operator << (ostream& os, pair<P, Q> p) { os << "(" << p.first << "," << p.second << ")"; return os; }
template<typename P, typename Q> istream& operator >> (istream& is, pair<P, Q>& p) { is >> p.first >> p.second; return is; }
template<typename T> ostream& operator << (ostream& os, vector<T> v) { os << "("; for (auto& i: v) os << i << ","; os << ")"; return os; }
template<typename T> istream& operator >> (istream& is, vector<T>& v) { for (auto& i: v) is >> i; return is; }
template<typename T> ostream& operator << (ostream& os, set<T> s) { os << "#{"; for (auto& i: s) os << i << ","; os << "}"; return os; }
template<typename K, typename V> ostream& operator << (ostream& os, map<K, V> m) { os << "{"; for (auto& i: m) os << i << ","; os << "}"; return os; }
template<typename E, size_t N> istream& operator >> (istream& is, array<E, N>& a) { for (auto& i: a) is >> i; return is; }
template<typename E, size_t N> ostream& operator << (ostream& os, array<E, N>& a) { os << "[" << N << "]{"; for (auto& i: a) os << i << ","; os << "}"; return os; }

template<typename T> inline T setmax(T& a, T b) { return a = std::max(a, b); }
template<typename T> inline T setmin(T& a, T b) { return a = std::min(a, b); }

__attribute__((constructor)) static void ___initio(void) { ios_base::sync_with_stdio(false); cin.tie(nullptr); cout.setf(ios_base::fixed); cout.precision(15); return ; }

using lli = long long int;
using ull = unsigned long long;
using str = string;
template<typename T> using vec = vector<T>;

constexpr array<int, 8> di({0, 1, -1, 0, 1, -1, 1, -1});
constexpr array<int, 8> dj({1, 0, 0, -1, 1, -1, -1, 1});
constexpr lli mod = 1e9 + 7;
// constexpr lli mod = 998244353;

vec<pair<lli, lli>> fn(lli x, lli w)
{
  vec<pair<lli, lli>> v;
  for (lli i = 2; i * i <= x; ++i) {
    if (x % i == 0) {
      int a = 0;
      while (x % i == 0) {
        ++a;
        x /= i;
      }
      v.push_back(make_pair(i, a * w % mod));
    }
  }
  if (x != 1) v.push_back(make_pair(x, w));
  return v;
}

vec<pair<lli, lli>> merge(vec<pair<lli, lli>> a, vec<pair<lli, lli>> b)
{
  sort(a.begin(), a.end());
  sort(b.begin(), b.end());
  vec<pair<lli, lli>> v;
  int i = 0;
  int j = 0;
  while (i < a.size() && j < b.size()) {
    if (a[i].first == b[j].first) {
      v.push_back(make_pair(a[i].first, (a[i].second + b[j].second) % mod));
      ++i;
      ++j;
    } else if (a[i].first < b[j].first) {
      v.push_back(a[i]);
      ++i;
    } else {
      v.push_back(b[j]);
      ++j;
    }
  }
  while (i < a.size()) v.push_back(a[i++]);
  while (j < b.size()) v.push_back(b[j++]);
  return v;
}

  lli mod_pow(lli n, lli p, lli M = mod)
  {
    if (p == 0) return 1;
    if (p == 1) return n % M;
    lli m = mod_pow(n, p / 2, M);
    m *= m;
    m %= M;
    if (p % 2) m = (m * n) % M;
    return m;
  }

int main(int argc, char *argv[])
{
  lli n, k;
  while (cin >> n >> k) {
    vec<pair<lli, lli>> u = fn(n, 1);
    // cout << u << endl;
    while (k--) {
      vec<pair<lli, lli>> v = u;
      u.clear();
      each (i, v) {
        u = merge(u, fn(i.first + 1, i.second));
      }
      // cout << u << endl; each (i, u) assert(i.second);
      if (u.size() == 1 && u.front().first == 2) break;
      if (u.size() == 2 && u[0].first == 2 && u[1].first == 3) break;
    }
    if (u.size() == 2 && u[0].first == 2 && u[1].first == 3) {
      lli a = mod_pow(u[0].second, k / 2);
      lli b = mod_pow(u[1].second, k / 2);
      // cout << u << ' ' << k << ' ' << p << endl;
      u[0].second = u[0].second * a;
      u[1].second = u[1].second * b;
      if (k % 2) {
        u[0].first = 3;
        u[1].first = 2;
      }
    } else if (u.size() == 1 && u.front().first == 2 && k) {
      lli p = mod_pow(u[0].second, k / 2);
      // cout << u << ' ' << k << ' ' << p << endl;
      u[0].second = u[0].second * p;
      if (k % 2) {
        u[0].first = 3;
      }
    }
    lli z = 1;
    each (i, u) {
      z *= mod_pow(i.first, i.second);
      z %= mod;
    }
    cout << z << endl;
  }

  return 0;
}
0