結果

問題 No.368 LCM of K-products
ユーザー asi1024
提出日時 2016-05-22 14:21:21
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 110 ms / 2,000 ms
コード長 880 bytes
コンパイル時間 1,868 ms
コンパイル使用メモリ 184,512 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-22 14:47:41
合計ジャッジ時間 3,165 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>

#define REP(i,n) for(int i=0;i<(int)(n);i++)
#define ALL(x) (x).begin(),(x).end()

using namespace std;

using ll = long long;

vector<pair<int,int>> prime_factorization(int n) {
  vector<pair<int,int>> res;
  for (int i = 2; i * i <= n; ++i) {
    int cnt = 0;
    while (n % i == 0) { ++cnt; n /= i; }
    if (cnt > 0) res.emplace_back(i, cnt);
  }
  if (n > 1) res.emplace_back(n, 1);
  return res;
}

const int mod = 1e9 + 7;

int main() {
  int N, K;
  cin >> N >> K;
  map<int,vector<int>> vec;
  REP(i,N) {
    int a; cin >> a;
    for (auto p: prime_factorization(a))
      vec[p.first].push_back(p.second);
  }
  ll res = 1;
  for (auto p: vec) {
    int i = p.first;
    vector<int> v = p.second;
    sort(v.rbegin(), v.rend());
    REP(j,min((int)v.size(), K))
      REP(k,v[j]) res = (res * i) % mod;
  }
  cout << res << endl;
  return 0;
}
0