結果
問題 | No.1917 LCMST |
ユーザー |
![]() |
提出日時 | 2023-06-29 00:05:55 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 733 ms / 4,000 ms |
コード長 | 3,193 bytes |
コンパイル時間 | 1,994 ms |
コンパイル使用メモリ | 181,224 KB |
実行使用メモリ | 94,716 KB |
最終ジャッジ日時 | 2024-07-05 17:26:37 |
合計ジャッジ時間 | 25,959 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 42 |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; using vec = vector<ll>; using mat = vector<vec>; using pll = pair<ll,ll>; using dvec = vector<double>; using dmat = vector<dvec>; #define INF (1LL<<61) #define MOD 1000000007LL //#define MOD 998244353LL #define PR(x) cout << (x) << endl #define PS(x) cout << (x) << " " #define REP(i,m,n) for(ll (i)=(m),(i_len)=(n);(i)<(i_len);++(i)) #define FORE(i,v) for(auto (i):v) #define ALL(x) (x).begin(), (x).end() #define SZ(x) ((ll)(x).size()) #define REV(x) reverse(ALL((x))) #define ASC(x) sort(ALL((x))) #define DESC(x) {ASC((x)); REV((x));} #define BIT(s,i) (((s)>>(i))&1) #define pb push_back #define fi first #define se second class mint { public: ll x; mint(ll x=0) : x((x%MOD+MOD)%MOD) {} mint operator-() const {return mint(-x);} mint& operator+=(const mint& a) {if((x+=a.x)>=MOD) x-=MOD; return *this;} mint& operator-=(const mint& a) {if((x+=MOD-a.x)>=MOD) x-=MOD; return *this;} mint& operator*=(const mint& a) {(x*=a.x)%=MOD; return *this;} mint operator+(const mint& a) const {mint b(*this); return b+=a;} mint operator-(const mint& a) const {mint b(*this); return b-=a;} mint operator*(const mint& a) const {mint b(*this); return b*=a;} mint pow(ll t) const {if(!t) return 1; mint a=pow(t>>1); return (t&1?*this*a:a)*a;} mint inv() const {return pow(MOD-2);} mint& operator/=(const mint& a) {return *this*=a.inv();} mint operator/(const mint& a) const {mint b(*this); return b/=a;} }; istream &operator>>(istream& is, mint& a) {ll t; is>>t; a=t; return is;} ostream &operator<<(ostream& os, const mint& a) {return os<<a.x;} using mvec = vector<mint>; using mmat = vector<mvec>; template<class T> inline int chmin(T& a, T b) {if(a>b) {a=b; return 1;} return 0;} template<class T> inline int chmax(T& a, T b) {if(a<b) {a=b; return 1;} return 0;} struct UnionFind { vec par; vec rank; vec num; UnionFind(ll N) { par = vec(N); REP(i,0,N) par[i] = i; rank = vec(N, 0); num = vec(N, 1); } ll root(ll x) { if(par[x] == x) return x; return par[x] = root(par[x]); } void unite(ll x, ll y) { ll rx = root(x); ll ry = root(y); if(rx == ry) return; if(rank[rx] < rank[ry]) swap(rx, ry); par[ry] = rx; num[rx] += num[ry]; if(rank[rx] == rank[ry]) ++rank[rx]; } bool same(ll x, ll y) { return root(x) == root(y); } ll size(ll x) { return num[root(x)]; } }; int main() { ll N; cin >> N; vec A(N); REP(i,0,N) cin >> A[i]; ll ans = 0, M = 100010; vec C(M), D(M, -1); FORE(i,A) { if(C[i] == 0) C[i] = 1; else ans += i; } mat E; REP(i,1,M) { for(ll j=i; j<M; j+=i) { if(C[j] == 1) { if(D[i] == -1) D[i] = j; else E.pb({D[i]*j/i, D[i], j}); } } } ASC(E); UnionFind uf(M); FORE(e,E) { if(!uf.same(e[1], e[2])) { ans += e[0]; uf.unite(e[1], e[2]); } } PR(ans); return 0; } /* */