#421125 (C++14) No.36 素数が嫌い！

• 問題: No.36 素数が嫌い！
• ユーザー: wunderkammer2
• 提出日時: 2020-01-24 16:47:57
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,284 bytes
• コンパイル時間: 886 ms
• コンパイル使用メモリ: 99,488 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-09-14 03:43:51
• 合計ジャッジ時間: 2,269 ms
• ジャッジサーバーID （参考情報）: judge2 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	AC	4 ms; 5,376 KB
testcase_02	WA	-
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	WA	-
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	WA	-
testcase_12	WA	-
testcase_13	AC	102 ms; 5,376 KB
testcase_14	AC	2 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	WA	-
testcase_17	AC	2 ms; 5,376 KB
testcase_18	AC	2 ms; 5,376 KB
testcase_19	AC	6 ms; 5,376 KB
testcase_20	AC	5 ms; 5,376 KB
testcase_21	AC	15 ms; 5,376 KB
testcase_22	AC	3 ms; 5,376 KB
testcase_23	AC	2 ms; 5,376 KB
testcase_24	AC	16 ms; 5,376 KB
testcase_25	WA	-
testcase_26	AC	37 ms; 5,376 KB
testcase_27	AC	3 ms; 5,376 KB
testcase_28	AC	36 ms; 5,376 KB
testcase_29	AC	2 ms; 5,376 KB

ソースコード

#include<algorithm>
#include<cmath>
#include<cstdio>
#include<functional>
#include<iomanip>
#include<iostream>
#include<map>
#include<queue>
#include<set>
#include<string>
#include<utility>
#include<vector>

using namespace std;
typedef long long ll;
const ll mod = 1000000007;
#define rep(i,n) for(int i=0;i<n;i++)
#define repl(i,s,e) for(int i=s;i<e;i++)
#define reple(i,s,e) for(int i=s;i<=e;i++)
#define revrep(i,n) for(int i=n-1;i>=0;i--)
#define all(x) (x).begin(),(x).end()

int main()
{	
	ll N;
	cin >> N;

	if (N == 1)
	{
		cout << "NO" << endl;
		return 0;
	}

	map<ll, ll> counts;

	for (ll i = 2; i * i <= N; i++)
	{
		//最大回数まで割るので、結果的に素数でしか割れない
		while (N % i == 0)
		{
			counts[i]++;
			N /= i;
		}
	}


	//素因数で合成数を作れるか確認
	//素因数が１種類のみの場合
	if (counts.size() == 1)
	{
		//1乗は、素数なのでNG
		//2乗は、1、素因数p、N（p^2）となるためNG
		//3乗は、1、素因数p、p^2、N（p^3）となるためOK
		for (auto c : counts)
		{
			if (c.second >= 3)
			{
				cout << "YES" << endl;
				return 0;
			}
			else
			{
				cout << "NO" << endl;
				return 0;
			}
		}
	}
	
	//素因数が複数種類ある時はOK
	cout << "YES" << endl;

	return 0;
}