結果

問題 No.368 LCM of K-products
ユーザー tnakao0123tnakao0123
提出日時 2016-07-15 11:18:43
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 2,465 bytes
コンパイル時間 2,070 ms
コンパイル使用メモリ 100,284 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-15 01:04:38
合計ジャッジ時間 2,371 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 368.cc: No.368 LCM of K-products - yukicoder
 */

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<deque>
#include<algorithm>
#include<numeric>
#include<utility>
#include<complex>
#include<functional>
 
using namespace std;

/* constant */

const int MAX_N = 1000;
const int MAX_K = 1000;
const int MAX_P = 31623; // >=sqrt(10^9)

typedef long long ll;
const ll MOD = 1000000007;

/* typedef */

typedef vector<int> vi;
typedef pair<int,int> pii;
typedef vector<pii> vpii;
typedef map<int,vi> mivi;

/* global variables */

bool primes[MAX_P + 1];
int as[MAX_N];

/* subroutines */

void gen_primes(int maxp, vi &pnums) {
  memset(primes, true, sizeof(primes));
  primes[0] = primes[1] = false;

  int p;
  for (p = 2; p * p <= maxp; p++)
    if (primes[p]) {
      pnums.push_back(p);
      for (int q = p * p; q <= maxp; q += p) primes[q] = false;
    }
  for (; p <= maxp; p++)
    if (primes[p]) pnums.push_back(p);
}

bool prime_decomp(int n, vi &pnums, vpii& pds) {
  pds.clear();

  int pn = pnums.size();
  for (int i = 0; i < pn; i++) {
    int pi = pnums[i];
    if (pi * pi > n) {
      if (n > 1) pds.push_back(pii(n, 1));
      return true;
    }

    if (n % pi == 0) {
      int fi = 0;
      while (n % pi == 0) n /= pi, fi++;
      pds.push_back(pii(pi, fi));
    }
  }
  return false;
}

ll powmod(ll a, int b) {
  ll pm = 1;
  while (b > 0) {
    if ((b & 1) != 0) pm = (pm * a) % MOD;
    a = (a * a) % MOD;
    b >>= 1;
  }
  return pm;
}

/* main */

int main() {
  vi pnums;
  gen_primes(MAX_P, pnums);

  int n, k;
  cin >> n >> k;

  mivi pvs;
  
  for (int i = 0; i < n; i++) {
    int ai;
    cin >> ai;

    vpii pds;
    prime_decomp(ai, pnums, pds);

    for (vpii::iterator vit = pds.begin(); vit != pds.end(); vit++)
      pvs[vit->first].push_back(vit->second);
  }

  ll lcm = 1;
  for (mivi::iterator mit = pvs.begin(); mit != pvs.end(); mit++) {
    const int &pi = mit->first;
    vi &v = mit->second;
    sort(v.begin(), v.end(), greater<int>());

    //printf("%d:", pi);
    //for (int i = 0; i < v.size(); i++) printf(" %d", v[i]); putchar('\n');
    
    int sum = 0, maxi = min(k, (int)v.size());
    for (int i = 0; i < maxi; i++) sum += v[i];
    lcm = (lcm * powmod((ll)pi, sum)) % MOD;
  }

  printf("%lld\n", lcm);
  return 0;
}
0