結果
問題 | No.1917 LCMST |
ユーザー |
![]() |
提出日時 | 2022-04-29 22:28:26 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 513 ms / 4,000 ms |
コード長 | 3,519 bytes |
コンパイル時間 | 2,895 ms |
コンパイル使用メモリ | 211,240 KB |
最終ジャッジ日時 | 2025-01-28 23:13:44 |
ジャッジサーバーID (参考情報) |
judge1 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 42 |
ソースコード
#define _USE_MATH_DEFINES #include <bits/stdc++.h> using namespace std; #define FOR(i,m,n) for(int i=(m);i<(n);++i) #define REP(i,n) FOR(i,0,n) #define ALL(v) (v).begin(),(v).end() using ll = long long; constexpr int INF = 0x3f3f3f3f; constexpr long long LINF = 0x3f3f3f3f3f3f3f3fLL; constexpr double EPS = 1e-8; constexpr int MOD = 1000000007; // constexpr int MOD = 998244353; constexpr int DY4[]{1, 0, -1, 0}, DX4[]{0, -1, 0, 1}; constexpr int DY8[]{1, 1, 0, -1, -1, -1, 0, 1}; constexpr int DX8[]{0, -1, -1, -1, 0, 1, 1, 1}; template <typename T, typename U> inline bool chmax(T& a, U b) { return a < b ? (a = b, true) : false; } template <typename T, typename U> inline bool chmin(T& a, U b) { return a > b ? (a = b, true) : false; } struct IOSetup { IOSetup() { std::cin.tie(nullptr); std::ios_base::sync_with_stdio(false); std::cout << fixed << setprecision(20); } } iosetup; std::vector<int> prime_sieve(const int n, const bool get_only_prime) { std::vector<int> smallest_prime_factor(n + 1), prime; std::iota(smallest_prime_factor.begin(), smallest_prime_factor.end(), 0); for (int i = 2; i <= n; ++i) { if (smallest_prime_factor[i] == i) prime.emplace_back(i); for (const int p : prime) { if (i * p > n || p > smallest_prime_factor[i]) break; smallest_prime_factor[i * p] = p; } } return get_only_prime ? prime : smallest_prime_factor; } struct Divisor { const std::vector<int> smallest_prime_factor; explicit Divisor(const int n) : smallest_prime_factor(prime_sieve(n, false)) {} std::vector<int> query(int n) const { std::vector<int> res{1}; while (n > 1) { const int prime_factor = smallest_prime_factor[n], d = res.size(); int tmp = 1; for (; n % prime_factor == 0; n /= prime_factor) { tmp *= prime_factor; for (int i = 0; i < d; ++i) { res.emplace_back(res[i] * tmp); } } } // std::sort(res.begin(), res.end()); return res; } }; struct UnionFind { explicit UnionFind(const int n) : data(n, -1) {} int root(const int ver) { return data[ver] < 0 ? ver : data[ver] = root(data[ver]); } bool unite(int u, int v) { u = root(u); v = root(v); if (u == v) return false; if (data[u] > data[v]) std::swap(u, v); data[u] += data[v]; data[v] = u; return true; } bool is_same(const int u, const int v) { return root(u) == root(v); } int size(const int ver) { return -data[root(ver)]; } private: std::vector<int> data; }; int main() { constexpr int A = 100000; int n; cin >> n; vector<int> a(n); REP(i, n) cin >> a[i]; sort(ALL(a)); ll ans = 0; FOR(i, 1, n) { if (a[i - 1] == a[i]) ans += a[i]; } a.erase(unique(ALL(a)), a.end()); n = a.size(); Divisor divisor(A); vector<int> pos[A + 1]{}; REP(i, n) for (int j : divisor.query(a[i])) pos[j].emplace_back(i); priority_queue<pair<ll, int>, vector<pair<ll, int>>, greater<pair<ll, int>>> que; REP(i, A + 1) { if (pos[i].size() >= 2) { reverse(pos[i].begin() + 1, pos[i].end()); que.emplace(1LL * a[pos[i].front()] * a[pos[i].back()] / i, i); } } UnionFind union_find(n); while (!que.empty()) { const int i = que.top().second; que.pop(); if (union_find.unite(pos[i].front(), pos[i].back())) ans += lcm(1LL * a[pos[i].front()], a[pos[i].back()]); pos[i].pop_back(); if (pos[i].size() >= 2) que.emplace(1LL * a[pos[i].front()] * a[pos[i].back()] / i, i); } cout << ans << '\n'; return 0; }