結果
問題 |
No.3270 No Coprime Cycles
|
ユーザー |
![]() |
提出日時 | 2025-08-04 20:36:10 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 682 ms / 2,000 ms |
コード長 | 3,424 bytes |
コンパイル時間 | 5,627 ms |
コンパイル使用メモリ | 334,620 KB |
実行使用メモリ | 86,520 KB |
最終ジャッジ日時 | 2025-08-18 01:21:14 |
合計ジャッジ時間 | 32,671 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 42 |
ソースコード
#include <bits/stdc++.h> using namespace std; using std::cerr; using std::cin; using std::cout; #if __has_include(<atcoder/all>) #include <atcoder/all> using mint = atcoder::modint998244353; istream &operator>>(istream &is, mint &a) { long long t; is >> t; a = t; return is; } ostream &operator<<(ostream &os, mint a) { return os << a.val(); } #endif typedef long double ld; #define long long long #define uint unsigned int #define ull unsigned long #define overload3(a, b, c, name, ...) name #define rep3(i, a, b) for (int i = (a); i < (b); i++) #define rep2(i, n) rep3(i, 0, n) #define rep1(n) rep2(__i, n) #define rep(...) overload3(__VA_ARGS__, rep3, rep2, rep1)(__VA_ARGS__) #define per3(i, a, b) for (int i = (b) - 1; i >= (a); i--) #define per2(i, n) per3(i, 0, n) #define per1(n) per2(__i, n) #define per(...) overload3(__VA_ARGS__, per3, per2, per1)(__VA_ARGS__) #define all(a) a.begin(), a.end() #define UNIQUE(a) \ sort(all(a)); \ a.erase(unique(all(a)), a.end()) #define sz(a) (int)a.size() #define vec vector #ifndef DEBUG #define cerr \ if (0) \ cerr // #undef assert // #define assert(...) void(0) #undef endl #define endl '\n' #endif template <typename T> ostream &operator<<(ostream &os, vector<T> a) { const int n = a.size(); rep(i, n) { os << a[i]; if (i + 1 != n) os << " "; } return os; } template <typename T, size_t n> ostream &operator<<(ostream &os, array<T, n> a) { rep(i, n) { os << a[i]; if (i + 1 != n) os << " "; } return os; } template <typename T> istream &operator>>(istream &is, vector<T> &a) { for (T &i : a) is >> i; return is; } template <typename T, typename S> bool chmin(T &x, S y) { if ((T)y < x) { x = (T)y; return true; } return false; } template <typename T, typename S> bool chmax(T &x, S y) { if (x < (T)y) { x = (T)y; return true; } return false; } template <typename T> void operator++(vector<T> &a) { for (T &i : a) ++i; } template <typename T> void operator--(vector<T> &a) { for (T &i : a) --i; } template <typename T> void operator++(vector<T> &a, int) { for (T &i : a) i++; } template <typename T> void operator--(vector<T> &a, int) { for (T &i : a) i--; } void solve() { constexpr int INF = 1e6 + 1; vec<bool> is_prime(INF, true); vec<vec<int>> pr(INF); vec<int> prime; for (int i = 2; i < INF; i++) { if (!is_prime[i]) continue; for (int k = i; k < INF; k += i) { is_prime[k] = false; pr[k].push_back(sz(prime)); } prime.push_back(i); } int n; cin >> n; vec<int> a(n); cin >> a; vec<tuple<int, int, int>> li; rep(i, n) { auto prai = pr[a[i]]; for (int pi : prai) { int v = 0; const int p = prime[pi]; while (a[i] % p == 0) { a[i] /= p; v += p; } li.emplace_back(v, i, n + pi); } } sort(all(li), greater<tuple<int, int, int>>()); atcoder::dsu dsu(n + sz(pr)); int ans = 0; for (auto [v, i, j] : li) { if (dsu.same(i, j)) { ans += v; continue; } dsu.merge(i, j); } cout << ans << endl; return; } int main() { // srand((unsigned)time(NULL)); cin.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(20); int t = 1; // cin >> t; while (t--) solve(); }