結果
問題 | No.1917 LCMST |
ユーザー |
|
提出日時 | 2023-01-27 02:03:22 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 485 ms / 4,000 ms |
コード長 | 1,373 bytes |
コンパイル時間 | 2,327 ms |
コンパイル使用メモリ | 204,644 KB |
最終ジャッジ日時 | 2025-02-10 07:17:52 |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 42 |
ソースコード
#include <bits/stdc++.h> using namespace std; #define fi first #define se second #define pb push_back using vi = vector <int>; using ll = long long; using pii = pair <int, int>; //~ const ll mod = 998244353; const ll mod = 1e9 + 7; ll qpow(ll a, ll b, ll m = mod) { ll r = 1, t = a; for(; b; b /= 2) { if(b & 1) r = r * t % m; t = t * t % m; } return r; } const int N = 1e5 + 11; vi divs[N]; vi muls[N]; int a[N]; int p[N]; int find(int x) { if(p[x] == x) return x; return p[x] = find(p[x]); } struct edge { int u, v; ll w; bool operator <(const edge &e) const { return w < e.w; } }; int main() { gcd(1, 2); ios :: sync_with_stdio(false); int n, m; cin >> n; ll ans = 0; for(int i = 1; i <= n; i ++) { int x; cin >> x; if(a[x]) ans += x; a[x] = 1; } for(int i = 1; i < N; i ++) for(int j = i; j < N; j += i) { divs[j].pb(i); if(a[j]) muls[i].pb(j); } for(int i = 1; i < N; i ++) if(muls[i].size()) sort(muls[i].begin(), muls[i].end()); vector <edge> edges; for(int i = 1; i < N; i ++) if(a[i]) { for(int x : divs[i]) { if(muls[x].size()) edges.pb({i, muls[x][0], lcm(ll(i), ll(muls[x][0]))}); } } sort(edges.begin(), edges.end()); iota(p, p + N, 0); for(auto e : edges) { int u = e.u, v = e.v; u = find(u); v = find(v); if(u != v) { ans += e.w; p[u] = v; } } cout << ans << '\n'; }