結果

問題 No.1917 LCMST
ユーザー rabimearabimea
提出日時 2023-01-27 01:22:43
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,596 bytes
コンパイル時間 2,558 ms
コンパイル使用メモリ 211,456 KB
実行使用メモリ 56,760 KB
最終ジャッジ日時 2024-06-27 15:12:58
合計ジャッジ時間 17,172 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 56 ms
16,512 KB
testcase_01 AC 57 ms
16,512 KB
testcase_02 AC 58 ms
16,512 KB
testcase_03 AC 64 ms
17,280 KB
testcase_04 AC 68 ms
17,280 KB
testcase_05 AC 64 ms
17,280 KB
testcase_06 AC 66 ms
17,280 KB
testcase_07 AC 63 ms
17,280 KB
testcase_08 AC 56 ms
16,512 KB
testcase_09 AC 370 ms
56,364 KB
testcase_10 AC 370 ms
56,492 KB
testcase_11 AC 365 ms
56,544 KB
testcase_12 AC 339 ms
39,436 KB
testcase_13 WA -
testcase_14 AC 348 ms
56,040 KB
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 AC 200 ms
21,788 KB
testcase_20 AC 183 ms
19,212 KB
testcase_21 AC 202 ms
21,960 KB
testcase_22 AC 184 ms
19,476 KB
testcase_23 AC 182 ms
19,212 KB
testcase_24 AC 201 ms
21,788 KB
testcase_25 AC 183 ms
19,528 KB
testcase_26 AC 188 ms
21,424 KB
testcase_27 AC 192 ms
21,684 KB
testcase_28 AC 194 ms
21,672 KB
testcase_29 AC 350 ms
56,636 KB
testcase_30 AC 353 ms
56,636 KB
testcase_31 AC 358 ms
56,636 KB
testcase_32 AC 354 ms
56,600 KB
testcase_33 AC 358 ms
56,632 KB
testcase_34 AC 150 ms
16,512 KB
testcase_35 AC 151 ms
16,512 KB
testcase_36 AC 151 ms
16,512 KB
testcase_37 AC 353 ms
56,760 KB
testcase_38 AC 358 ms
56,748 KB
testcase_39 AC 350 ms
56,628 KB
testcase_40 AC 360 ms
56,620 KB
testcase_41 AC 362 ms
56,628 KB
testcase_42 AC 366 ms
56,704 KB
testcase_43 WA -
testcase_44 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
 
#define fi first
#define se second
#define pb push_back
using vi = vector <int>;
using ll = long long;
using pii = pair <int, int>;
//~ const ll mod = 998244353;
const ll mod = 1e9 + 7;
ll qpow(ll a, ll b, ll m = mod) { ll r = 1, t = a;
	for(; b; b /= 2) { if(b & 1) r = r * t % m; t = t * t % m; } return r; }

const int N = 1e5 + 11;
vi divs[N];
vi muls[N];

int a[N];

int p[N];
int find(int x) {
	if(p[x] == x) return x;
	return p[x] = find(p[x]);
}

struct edge {
	int u, v;
	ll w;
	bool operator <(const edge &e) const {
		return w < e.w;
	}
};

ll lcm(ll a, ll b) {
	return a / __gcd(a, b) * b;
}

int main() {
	ios :: sync_with_stdio(false);
	
	int n, m; cin >> n;
	
	ll ans = 0;
	for(int i = 1; i <= n; i ++) {
		int x; cin >> x;
		if(a[x]) ans += x;
		a[x] = 1;
	}
	
	for(int i = 1; i < N; i ++)
		for(int j = i; j < N; j += i) {
			divs[j].pb(i);
			if(a[j]) muls[i].pb(j);
		}
	for(int i = 1; i < N; i ++)
		sort(muls[i].begin(), muls[i].end());
	
	vector <edge> edges;
	//~ for(int i = 1; i < N; i ++) if(a[i]) {
		//~ for(int x : divs[i]) {
			//~ if(muls[x].size())
				//~ edges.pb({i, muls[x][0], lcm(i, muls[x][0])});
		//~ }
	//~ }
	
	for(int i = 1; i < N; i ++) if(muls[i].size()) {
		int x = muls[i][0];
		for(int y : muls[i])
			edges.pb({x, y, lcm(x, y)});
	}
	
	sort(edges.begin(), edges.end());
	
	iota(p, p + N, 0);
	for(auto e : edges) {
		//~ cout << e.u << ' ' << e.v << ' ' << e.w << '\n';
		int u = e.u, v = e.v;
		u = find(u); v = find(v);
		if(u != v) {
			ans += e.w;
			p[u] = v;
		}
	}
	
	cout << ans << '\n';
}
0