結果

問題 No.368 LCM of K-products
ユーザー fine
提出日時 2017-11-02 01:46:06
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,308 bytes
コンパイル時間 2,573 ms
コンパイル使用メモリ 179,932 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-22 14:57:36
合計ジャッジ時間 3,353 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

using ll = long long;

const ll MOD = 1e9 + 7;

ll powmod(int x, int n) {
    ll res = 1;
    while (n > 0) {
        if (n & 1) res = res * x % MOD;
        x = x * x % MOD;
        n >>= 1;
    }
    return res;
}

void add_memo(map<int, vector<int> >& memo, int x, int cnt) {
    if (memo.find(x) == memo.end()) {
        memo[x] = vector<int>();
    }
    memo[x].push_back(cnt);
}

int a[1005];

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    int N, K;
    cin >> N >> K;
    for (int i = 0; i < N; i++) {
        cin >> a[i];
    }
    ll ans = 1;
    map<int, vector<int> > memo;
    for (int i = 0; i < N; i++) {
        for (int j = 2; j * j <= a[i]; j++) {
            int cnt = 0;
            while (a[i] % j == 0) {
                a[i] /= j;
                cnt++;
            }
            if (cnt == 0) continue;
            add_memo(memo, j, cnt);
        }
        if (a[i] != 1) add_memo(memo, a[i], 1);       
    }
    for (auto& p : memo) {
        sort(p.second.rbegin(), p.second.rend());
        int sum = 0, sz = min((int)p.second.size(), K);
        for (int i = 0; i < sz; i++) {
            sum += p.second[i];
        }
        (ans *= powmod(p.first, sum)) %= MOD;
    }
    cout << ans << endl;
    return 0;
}
0