結果

問題 No.1917 LCMST
ユーザー 👑 potato167
提出日時 2022-04-29 23:51:12
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 444 ms / 4,000 ms
コード長 2,746 bytes
コンパイル時間 2,453 ms
コンパイル使用メモリ 217,952 KB
最終ジャッジ日時 2025-01-29 00:37:52
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#define _GLIBCXX_DEBUG
using namespace std;
using std::cout;
using std::cin;
using std::endl;
using ll=long long;
using ld=long double;
ll ILL=2167167167167167167;
const int INF=2100000000;
const ll mod=998244353;
#define rep(i,a) for (ll i=0;i<a;i++)
#define all(p) p.begin(),p.end()
template<class T> using _pq = priority_queue<T, vector<T>, greater<T>>;
template<class T> ll LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> ll UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> bool chmin(T &a,const T &b){if(a>b){a=b;return 1;}else return 0;}
template<class T> bool chmax(T &a,const T &b){if(a<b){a=b;return 1;}else return 0;}
template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());}
template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});}
void yneos(bool a){if(a) cout<<"Yes\n"; else cout<<"No\n";}
template<class T> void vec_out(vector<T> &p){for(int i=0;i<(int)(p.size());i++){if(i) cout<<" ";cout<<p[i];}cout<<"\n";}

namespace po167{
struct UFtree
{
	std::vector<int> wei;
	std::vector<int> q;
	int component;
	UFtree(int n):wei(n),component(n),par(n){
		for(int i=0;i<n;i++){
			wei[i]=1,par[i]=i;
		}
	}
	void intialize(){
		for(auto x:q){
			wei[root(x)]=1;
			par[x]=x;
		}
		component=(int)par.size();
		q={};
	}
	//根っこを返す
	int root(int a){
		if(a==par[a]) return a;
		return par[a]=root(par[a]);
	}
	//trueなら1,falseなら0
	int same(int a,int b){
		if(root(a)==root(b)) return 1;
		else return 0;
	}
	//a,bが違う根っこの元なら結合する,結合したらtrueを返す
	bool unite(int a,int b){
		a=root(a),b=root(b);
		if(a==b) return false;
		if(wei[a]<wei[b]) std::swap(a,b);
		par[b]=a;
		q.push_back(b);
		wei[a]+=wei[b];
		component--;
		return true;
	}
	private:
	std::vector<int> par;
};
}
using po167::UFtree;


void solve();
// oddloop
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	
	int t=1;
	//cin>>t;
	rep(i,t) solve();
}

void solve(){
	int N;
	cin>>N;
	vector<ll> A(N);
	rep(i,N) cin>>A[i];
	int L=1e5+10;
	vector<bool> p(L);
	vector<ll> q(L,INF);
	vector<vector<int>> table(L);
	rep(i,L){
		if(i==0) continue;
		for(int j=i;j<L;j+=i) table[j].push_back(i);
	}
	ll ans=0;
	rep(i,N){
		if(p[A[i]]) ans+=A[i];
		for(auto x:table[A[i]]){
			chmin(q[x],A[i]);
		}
		p[A[i]]=1;
	}
	UFtree T(L);
	vector<tuple<ll,int,int>> e;
	rep(i,L){
		if(!p[i]) continue;
		for(auto x:table[i]){
			if(q[x]==INF) continue;
			e.push_back({q[x]*i/x,q[x],i});
		}
	}
	So(e);
	for(auto x:e){
		ll a,b,c; 
		tie(a,b,c)=x;
		if(T.unite(b,c)) ans+=a;
		//cout<<a<<"<"<<b<<"<"<<c<<"<"<<endl;
	}
	cout<<ans<<"\n";
}
0