#890682 (C++17) No.2379 Burnside's Theorem

• 問題: No.2379 Burnside's Theorem
• ユーザー: achapi
• 提出日時: 2023-07-14 21:22:13
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 2 ms / 2,000 ms
• コード長: 1,666 bytes
• コンパイル時間: 2,343 ms
• コンパイル使用メモリ: 211,756 KB
• 実行使用メモリ: 4,352 KB
• 最終ジャッジ日時: 2023-10-14 11:16:40
• 合計ジャッジ時間: 3,259 ms
• ジャッジサーバーID （参考情報）: judge12 / judge13

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 4,352 KB
testcase_01	AC	1 ms; 4,348 KB
testcase_02	AC	2 ms; 4,348 KB
testcase_03	AC	1 ms; 4,348 KB
testcase_04	AC	2 ms; 4,352 KB
testcase_05	AC	2 ms; 4,352 KB
testcase_06	AC	1 ms; 4,348 KB
testcase_07	AC	1 ms; 4,348 KB
testcase_08	AC	2 ms; 4,348 KB
testcase_09	AC	2 ms; 4,348 KB
testcase_10	AC	1 ms; 4,352 KB
testcase_11	AC	1 ms; 4,352 KB
testcase_12	AC	1 ms; 4,348 KB
testcase_13	AC	2 ms; 4,348 KB
testcase_14	AC	2 ms; 4,348 KB
testcase_15	AC	1 ms; 4,348 KB
testcase_16	AC	2 ms; 4,352 KB
testcase_17	AC	1 ms; 4,352 KB
testcase_18	AC	1 ms; 4,352 KB
testcase_19	AC	1 ms; 4,352 KB
testcase_20	AC	2 ms; 4,352 KB
testcase_21	AC	1 ms; 4,348 KB
testcase_22	AC	1 ms; 4,348 KB
testcase_23	AC	2 ms; 4,352 KB

ソースコード

#include <bits/stdc++.h>
using namespace std;

template<class T> T pow_mod(T A, T N, T M) {
	T res = 1 % M;
	A %= M;
	while (N) {
		if (N & 1) res = (res * A) % M;
		A = (A * A) % M;
		N >>= 1;
	}
	return res;
}

bool is_prime(long long N) {
	if (N <= 1) return false;
	if (N == 2 || N == 3) return true;
	if (N % 2 == 0) return false;
	vector<long long> A = {2, 325, 9375, 28178, 450775, 9780504, 1795265022};
	long long s = 0, d = N - 1;
	while (d % 2 == 0) {
		++s;
		d >>= 1;
	}
	for (auto a : A) {
		if (a % N == 0) return true;
		long long t, x = pow_mod<__int128_t>(a, d, N);
		if (x != 1) {
			for (t = 0; t < s; ++t) {
				if (x == N - 1) break;
				x = __int128_t(x) * x % N;
			}
			if (t == s) return false;
		}
	}
	return true;
}	
long long pollard(long long N) {
	if (N % 2 == 0) return 2;
	if (is_prime(N)) return N;
	long long step = 0;
	auto f = [&](long long x) -> long long {
		return (__int128_t(x) * x + step) % N;
	};
	while (true) {
		++step;
		long long x = 1, y = f(x);
		while (true) {
			long long p = gcd(y - x + N, N);
			if (p == 0 || p == N) break;
			if (p != 1) return p;
			x = f(x);
			y = f(f(y));
		}
	}
}

vector<long long> prime_factorize(long long N) {
	if (N == 1) return {};
	long long p = pollard(N);
	if (p == N) return {p};
	vector<long long> left = prime_factorize(p);
	vector<long long> right = prime_factorize(N / p);
	left.insert(left.end(), right.begin(), right.end());
	sort(left.begin(), left.end());
	return left;
}

int main() {
	long long N;
	cin >> N;
	auto s = prime_factorize(N);
	set<long long> T(s.begin(), s.end());
	if ((int)T.size() <= 2){
		cout << "Yes" << '\n';
	} else {
		cout << "No" << '\n';
	}
}