結果

問題 No.1917 LCMST
ユーザー Hamza Monther
提出日時 2023-02-10 21:50:32
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,728 bytes
コンパイル時間 2,239 ms
コンパイル使用メモリ 202,104 KB
最終ジャッジ日時 2025-02-10 12:36:12
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 WA * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define ll long long
struct DSU {
    vector<int> Rank, Root, mn;
    DSU(int n) {
        Rank.resize(n + 1, 1);
        Root.resize(n + 1);
        iota(Root.begin(), Root.end(), 0);
        mn = Root;
    }
    int find(int Node) {
        return Root[Node] = (Node == Root[Node] ? Node : find(Root[Node]));
    }
    bool check(int a, int b) {
        return (find(a) == find(b));
    }
    void merge(int a, int b) {
        a = find(a);
        b = find(b);
        if (check(a, b)) return;
        if (Rank[a] > Rank[b]) swap(a, b);
        Root[a] = b;
        Rank[b] += Rank[a];
        mn[b] = min(mn[b], mn[a]);
    }
};
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    const int N = 1e5 + 5;
    vector<int> mx(N, 0);
    vector<int> counter(N, 0);
    int n;
    cin >> n;
    for (int i = 0, x; i < n; ++i) {
        cin >> x;
        counter[x]++;
    }
    ll ans = 0;
    int mn = 1e9;
    for (int i = 1; i < N; ++i) {
        if (counter[i] == 0) {
            continue;
        }
        for (int j = i + i; j < N; j += i) {
            if (counter[j] > 0) {
                mx[j] = i;
            }
        }
        ans += 1ll * (counter[i] - 1) * i;
        if (mn == 1e9) {
            mn = i;
        }
    }
    assert(mn != 1e9);
    DSU dsu(N);
    for (int i = 1; i < N; ++i) {
        if (counter[i] != 0 && mx[i] != 0) {
            dsu.merge(i, mx[i]);
            ans += i;
        }
    }
    for (int i = 1; i < N; ++i) {
        if (counter[i] > 0 && dsu.find(i) == i && i != mn) {
            int x = dsu.mn[dsu.find(i)];
            ans += 1ll * x * mn / __gcd(x, mn);
        }
    }
    cout << ans << '\n';
    return 0;
}
0