結果

問題 No.1917 LCMST
ユーザー Koi
提出日時 2025-07-15 20:20:42
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,417 bytes
コンパイル時間 4,374 ms
コンパイル使用メモリ 314,656 KB
実行使用メモリ 109,060 KB
最終ジャッジ日時 2025-07-15 20:21:18
合計ジャッジ時間 30,639 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 1 WA * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/dsu>
using namespace std;
using namespace atcoder;
// #define long long long long
int main() {
    const long long MX = 100000;
    vector<vector<long long>> L(MX + 1);
    for (long long i = 1; i <= MX; ++i) {
        long long x = i;
        while (x < MX) {
            L[x].push_back(i);
            x += i;
        }
    }

    long long N;
    cin >> N;
    vector<long long> A(N);

    for (long long i = 0; i < N; ++i) cin >> A[i];
    sort(A.begin(), A.end());
    A.erase(unique(A.begin(), A.end()), A.end());
    N = (int)A.size();
    unordered_map<long long, vector<pair<long long, long long>>> D;
    vector<tuple<long long, long long, long long>> Mvec;

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

    for (auto &[key, vec] : D) {
    	if(vec.size() > 1){
	        sort(vec.begin(), vec.end());
	        for (long long 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