結果

問題 No.2756 GCD Teleporter
ユーザー kwm_tkwm_t
提出日時 2024-05-12 02:00:04
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 157 ms / 2,000 ms
コード長 3,001 bytes
コンパイル時間 6,063 ms
コンパイル使用メモリ 326,216 KB
実行使用メモリ 32,912 KB
最終ジャッジ日時 2024-05-12 02:00:16
合計ジャッジ時間 9,985 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
6,812 KB
testcase_01 AC 4 ms
6,940 KB
testcase_02 AC 3 ms
6,944 KB
testcase_03 AC 3 ms
6,940 KB
testcase_04 AC 13 ms
6,944 KB
testcase_05 AC 51 ms
10,624 KB
testcase_06 AC 134 ms
21,464 KB
testcase_07 AC 108 ms
17,888 KB
testcase_08 AC 97 ms
16,896 KB
testcase_09 AC 89 ms
15,104 KB
testcase_10 AC 137 ms
19,716 KB
testcase_11 AC 157 ms
23,912 KB
testcase_12 AC 31 ms
8,320 KB
testcase_13 AC 153 ms
24,776 KB
testcase_14 AC 140 ms
23,080 KB
testcase_15 AC 33 ms
8,448 KB
testcase_16 AC 76 ms
14,900 KB
testcase_17 AC 94 ms
17,152 KB
testcase_18 AC 32 ms
8,320 KB
testcase_19 AC 154 ms
25,280 KB
testcase_20 AC 107 ms
23,308 KB
testcase_21 AC 51 ms
12,928 KB
testcase_22 AC 135 ms
24,188 KB
testcase_23 AC 16 ms
6,944 KB
testcase_24 AC 85 ms
19,568 KB
testcase_25 AC 93 ms
20,792 KB
testcase_26 AC 103 ms
23,188 KB
testcase_27 AC 116 ms
24,196 KB
testcase_28 AC 76 ms
14,124 KB
testcase_29 AC 113 ms
18,968 KB
testcase_30 AC 135 ms
23,576 KB
testcase_31 AC 58 ms
12,416 KB
testcase_32 AC 54 ms
23,156 KB
testcase_33 AC 24 ms
11,264 KB
testcase_34 AC 14 ms
6,940 KB
testcase_35 AC 71 ms
23,176 KB
testcase_36 AC 88 ms
22,796 KB
testcase_37 AC 4 ms
6,944 KB
testcase_38 AC 106 ms
23,560 KB
testcase_39 AC 154 ms
32,912 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
//using mint = modint1000000007;
//const int mod = 1000000007;
//using mint = modint998244353;
//const int mod = 998244353;
//const int INF = 1e9;
const long long LINF = 1e18;
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define rep2(i,l,r)for(int i=(l);i<(r);++i)
#define rrep(i, n) for (int i = (n) - 1; i >= 0; --i)
#define rrep2(i,l,r)for(int i=(r) - 1;i>=(l);--i)
#define all(x) (x).begin(),(x).end()
#define allR(x) (x).rbegin(),(x).rend()
#define P pair<int,int>
template<typename A, typename B> inline bool chmax(A & a, const B & b) { if (a < b) { a = b; return true; } return false; }
template<typename A, typename B> inline bool chmin(A & a, const B & b) { if (a > b) { a = b; return true; } return false; }
// 素因数分解
std::vector<std::pair<long long, int>> primeFactorization(long long n) {
	std::vector<std::pair<long long, int>> ret;
	for (long long i = 2; i*i <= n; i++) {
		int c = 0;
		while (n%i == 0) { // 素数で割り切れなくなるまで割っていく
			c++;//割った個数を配列に足す
			n /= i;
		}
		if (0 != c) ret.push_back({ i,c });
		if (1 == n) break;
	}
	if (1 != n) ret.push_back({ n,1 });
	return ret;
}
long long solve(vector<int>a) {
	auto list = primeFactorization(a[0]);
	long long ret = 0;
	rep2(i, 1, a.size()) {
		if (1 != gcd(a[0], a[i]))continue;
		else ret += list[0].first;
	}
	return ret;
}
class Osa_k {
public:
	Osa_k(int max = 1000006) {
		size = max + 1;
		osa_k.resize(size);
	}
	void init() {
		osa_k[1] = 1;
		for (int i = 2; i < size + 1; ++i) {
			if (0 != osa_k[i])continue;
			osa_k[i] = i;
			for (int j = i * 2; j <= size; j += i) {
				if (0 == osa_k[j]) osa_k[j] = i;
			}
		}
	}
	std::vector<int> osk(int n) {
		std::vector<int>ret;
		while (1 != n) {
			ret.push_back(osa_k[n]);
			n /= osa_k[n];
		}
		return ret;
	}
	std::vector<std::pair<int, int>>oskPair(int n) {
		map<int, int>mp;
		auto ps = osk(n);
		for (auto p : ps)mp[p]++;
		std::vector<std::pair<int, int>> result;
		for (auto[k, v] : mp) {
			result.emplace_back(k, v);
		}
		return result;
	}
private:
	int size;
	std::vector<int>osa_k;
};
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int n; cin >> n;
	long long ans = LINF;
	vector<int>a(n);
	rep(i, n)cin >> a[i];
	map<int, int>mp;
	Osa_k osk(200005);
	osk.init();
	std::vector<std::vector<std::pair<int, int>>> p(n);
	rep(i, n) {
		p[i] = osk.oskPair(a[i]);
		for (auto e : p[i]) {
			if (mp.count(e.first))continue;
			mp[e.first] = mp.size();
		}
	}
	if (0 == mp.size()) {
		cout << 2 * n << endl;
		return 0;
	}
	dsu uf(n + mp.size());
	rep(i, n) {
		for (auto e : p[i]) {
			uf.merge(i, n + mp[e.first]);
		}
	}
	auto g = uf.groups();
	auto mnPrime = (*mp.begin()).first;
	if (1 == g.size()) {
		cout << 0 << endl;
	}
	else {
		long long ans = LINF;
		chmin(ans, (long long)(g.size()) * 2);
		chmin(ans, (long long)(g.size() - 1) * mnPrime);
		cout << ans << endl;
	}
	return 0;
}
0