結果

問題 No.1917 LCMST
コンテスト
ユーザー Koi
提出日時 2025-07-15 19:25:42
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,156 bytes
コンパイル時間 3,935 ms
コンパイル使用メモリ 308,016 KB
実行使用メモリ 740,440 KB
最終ジャッジ日時 2025-07-15 19:27:51
合計ジャッジ時間 12,685 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19 WA * 21 MLE * 1 -- * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/dsu>
using namespace std;
using namespace atcoder;

int main() {
    const int MX = 100000;
    vector<vector<int>> L(MX + 1);
    for (int i = 1; i <= MX; ++i) {
        int x = i;
        while (x < MX) {
            L[x].push_back(i);
            x += i;
        }
    }

    int N;
    cin >> N;
    vector<int> A(N);
    for (int i = 0; i < N; ++i) cin >> A[i];

    unordered_map<int, vector<pair<int, int>>> D;
    vector<tuple<long long, int, int>> Mvec;

    for (int i = 0; i < N; ++i) {
        for (int p : L[A[i]]) {
            D[p].emplace_back(A[i], i);
        }
    }

    for (auto &[key, vec] : D) {
        sort(vec.begin(), vec.end());
        for (int i = 1; i < vec.size(); ++i) {
            long long cost = 1LL * vec[0].first * vec[i].first / key;
            Mvec.emplace_back(cost, vec[0].second, vec[i].second);
        }
    }

    sort(Mvec.begin(), Mvec.end());
    dsu DS(N);

    long long ans = 0;
    for (auto &[cost, i, j] : Mvec) {
        if (!DS.same(i, j)) {
            DS.merge(i, j);
            ans += cost;
        }
    }

    cout << ans << '\n';
    return 0;
}
0