結果

問題 No.368 LCM of K-products
ユーザー xuzijian629xuzijian629
提出日時 2018-10-31 21:35:08
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 743 ms / 2,000 ms
コード長 2,137 bytes
コンパイル時間 2,495 ms
コンパイル使用メモリ 222,560 KB
実行使用メモリ 132,992 KB
最終ジャッジ日時 2024-09-22 14:58:25
合計ジャッジ時間 8,145 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 254 ms
44,032 KB
testcase_01 AC 743 ms
132,992 KB
testcase_02 AC 4 ms
5,376 KB
testcase_03 AC 199 ms
31,616 KB
testcase_04 AC 325 ms
47,744 KB
testcase_05 AC 258 ms
66,176 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 1 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 104 ms
25,600 KB
testcase_14 AC 335 ms
52,864 KB
testcase_15 AC 448 ms
63,360 KB
testcase_16 AC 301 ms
49,536 KB
testcase_17 AC 189 ms
37,376 KB
testcase_18 AC 481 ms
66,432 KB
testcase_19 AC 23 ms
8,320 KB
testcase_20 AC 151 ms
32,512 KB
testcase_21 AC 11 ms
5,760 KB
testcase_22 AC 424 ms
60,928 KB
testcase_23 AC 2 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 2 ms
5,376 KB
testcase_29 AC 2 ms
5,376 KB
testcase_30 AC 2 ms
5,376 KB
testcase_31 AC 2 ms
5,376 KB
testcase_32 AC 2 ms
5,376 KB
testcase_33 AC 51 ms
14,208 KB
testcase_34 AC 42 ms
14,208 KB
権限があれば一括ダウンロードができます

ソースコード

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