結果
問題 |
No.3270 No Coprime Cycles
|
ユーザー |
![]() |
提出日時 | 2025-09-12 23:59:57 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 3,991 bytes |
コンパイル時間 | 4,608 ms |
コンパイル使用メモリ | 274,632 KB |
実行使用メモリ | 69,492 KB |
最終ジャッジ日時 | 2025-09-13 00:01:13 |
合計ジャッジ時間 | 33,711 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 1 WA * 10 TLE * 9 -- * 22 |
ソースコード
#define ATCODER #include <bit> #include <cstdint> #include <iostream> #include <algorithm> #include <vector> #include <string> #include <queue> #include <cassert> #include <unordered_map> #include <unordered_set> #include <queue> #include <math.h> #include <climits> #include <set> #include <map> #include <list> #include <iterator> #include <bitset> #include <chrono> #include <type_traits> using namespace std; using ll = long long; #define FOR(i, a, b) for(ll i=(a); i<(b);i++) #define REP(i, n) for(ll i=0; i<(n);i++) #define ROF(i, a, b) for(ll i=(b-1); i>=(a);i--) #define PER(i, n) for(ll i=n-1; i>=0;i--) #define VL vector<ll> #define VVL vector<vector<ll>> #define VP vector< pair<ll,ll> > #define VVP vector<vector<pair<ll,ll>>> #define all(i) begin(i),end(i) #define SORT(i) sort(all(i)) #define EXISTBIT(x,i) (((x>>i) & 1) != 0) #define MP(a,b) make_pair(a,b) #ifdef ATCODER #include <atcoder/all> using namespace atcoder; using mint = modint1000000007; using mint2 = modint998244353; #endif template<typename T = ll> vector<T> read(size_t n) { vector<T> ts(n); for (size_t i = 0; i < n; i++) cin >> ts[i]; return ts; } template<typename TV, const ll N> void read_tuple_impl(TV&) {} template<typename TV, const ll N, typename Head, typename... Tail> void read_tuple_impl(TV& ts) { get<N>(ts).emplace_back(*(istream_iterator<Head>(cin))); read_tuple_impl<TV, N + 1, Tail...>(ts); } template<typename... Ts> decltype(auto) read_tuple(size_t n) { tuple<vector<Ts>...> ts; for (size_t i = 0; i < n; i++) read_tuple_impl<decltype(ts), 0, Ts...>(ts); return ts; } template<typename T> T det2(array<T, 4> ar) { return ar[0] * ar[3] - ar[1] * ar[2]; } template<typename T> T det3(array<T, 9> ar) { return ar[0] * ar[4] * ar[8] + ar[1] * ar[5] * ar[6] + ar[2] * ar[3] * ar[7] - ar[0] * ar[5] * ar[7] - ar[1] * ar[3] * ar[8] - ar[2] * ar[4] * ar[6]; } template<typename T> bool chmax(T& tar, T src) { return tar < src ? tar = src, true : false; } template<typename T> bool chmin(T& tar, T src) { return tar > src ? tar = src, true : false; } template<typename T> void inc(vector<T>& ar) { for (auto& v : ar) v++; } template<typename T> void dec(vector<T>& ar) { for (auto& v : ar) v--; } template<typename T> vector<pair<T, int>> id_sort(vector<T>& a) { vector<T, int> res(a.size()); for (int i = 0; i < a.size(); i++)res[i] = MP(a[i], i); SORT(res); return res; } using val = pair<ll, ll>; using func = ll; val op(val a, val b) { return MP(a.first + b.first, a.second + b.second); } val e() { return MP(0LL, 0LL); } val mp(func f, val a) { return MP(f * a.second, a.second); } func comp(func f, func g) { if (g == -1)return f; else return g; } func id() { return -1; } val mp2(func f, val a) { if (f == -1) { return e(); } return MP(a.first + f * a.second, a.second); } func comp2(func f, func g) { if (f == -1) { return g; } if (g == -1) { return -1; } return f + g; } func id2() { return 0; } // Rook ll dxr[4] = { 1,0,-1,0 }; ll dyr[4] = { 0,1,0,-1 }; // Bishop ll dxb[4] = { -1,-1,1,1 }; ll dyb[4] = { -1,1,-1,1 }; // qween ll dxq[8] = { 0,-1,-1,-1,0,1,1,1 }; ll dyq[8] = { -1,-1,0,1,1,1,0,-1 }; void solve() { ll n; cin >> n; VL a = read(n); VL p; ll mx = 1000000; VL k(mx + 1); unordered_map<ll, ll> rp; k[0] = k[1] = 1; REP(i, mx + 1) { if (k[i])continue; rp[i] = p.size(); p.push_back(i); ll vv = i * 2; while (vv <= mx) { k[vv] = 1; vv += i; } } priority_queue<tuple<ll, ll, ll>> que; ll ans = 0; dsu d(n + p.size()); REP(i, n) { for (ll v = 2; v * v <= a[i]; v++) { if (a[i] % v)continue; while (a[i] % v)a[i] /= v; que.emplace(v, i, rp[v]); } } while (!que.empty()) { auto [c, x, y] = que.top(); que.pop(); if (d.same(x, y)) { ans += c; continue; } d.merge(x, y); } cout << ans; return; } int main() { ll t = 1; // cin >> t; while (t--) { solve(); } return 0; }