結果

問題 No.368 LCM of K-products
ユーザー xuzijian629
提出日時 2018-10-31 21:35:08
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 687 ms / 2,000 ms
コード長 2,137 bytes
コンパイル時間 2,075 ms
コンパイル使用メモリ 214,096 KB
最終ジャッジ日時 2025-01-06 15:17:50
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using i64 = int64_t;
using vi = vector<i64>;
using vvi = vector<vi>;
mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());
constexpr i64 MOD = 1e9 + 7;

i64 modpow(i64 a, i64 n, i64 mod) {
    if (n == 0) return 1;
    if (n % 2 == 0) {
        i64 t = modpow(a, n / 2, mod);
        return t * t % mod;
    }
    return a * modpow(a, n - 1, mod) % mod;
}

bool is_prime(i64 n, int k = 10) {
    if (n == 2) return true;
    if (n < 2 || n % 2 == 0) return false;
    i64 d = n - 1;
    while (d % 2 == 0) {
        d /= 2;
    }
    for (int i = 0; i < k; i++) {
        i64 a = rnd() % (n - 2) + 1;
        i64 t = d;
        i64 y = modpow(a, t, n);
        while (t != n - 1 && y != 1 && y != n - 1) {
            y = modpow(y, 2, n);
            t *= 2;
        }
        if (y != n - 1 && t % 2 == 0) {
            return false;
        }
    }
    return true;
}

int main() {
    int n, k;
    cin >> n >> k;
    vi as(n);
    for (int i = 0; i < n; i++) {
        cin >> as[i];
    }

    map<i64, int> cnt[n];
    set<i64> ps;
    for (int i = 0; i < n; i++) {
        i64 a = as[i];
        if (is_prime(a)) {
            cnt[i][a]++;
            ps.insert(a);
        } else {
            while (a > 1) {
                if (is_prime(a)) {
                    cnt[i][a]++;
                    ps.insert(a);
                    break;
                }
                for (int j = 2; j * j <= a; j++) {
                    int change = 0;
                    while (a % j == 0) {
                        change = 1;
                        cnt[i][j]++;
                        ps.insert(j);
                        a /= j;
                    }              
                    if (change) break;  
                }
            }
        }
    }

    i64 ans = 1;
    for (i64 p: ps) {
        vi vs;
        for (int i = 0; i < n; i++) {
            vs.push_back(cnt[i][p]);
        }
        sort(vs.begin(), vs.end(), greater<>());
        ans *= modpow(p, accumulate(vs.begin(), vs.begin() + k, 0), MOD);
        ans %= MOD;
    }

    cout << ans << endl;
}
0