#480292 (Python3) No.36 素数が嫌い！

• 問題: No.36 素数が嫌い！
• ユーザー: shobonvip
• 提出日時: 2020-05-10 20:32:43
• 言語: Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
• 結果: WA
• 実行時間: -
• コード長: 582 bytes
• コンパイル時間: 138 ms
• コンパイル使用メモリ: 12,672 KB
• 実行使用メモリ: 10,880 KB
• 最終ジャッジ日時: 2024-07-08 06:14:17
• 合計ジャッジ時間: 11,991 ms
• ジャッジサーバーID （参考情報）: judge5 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	AC	559 ms; 10,880 KB
testcase_02	AC	25 ms; 10,752 KB
testcase_03	AC	25 ms; 10,752 KB
testcase_04	AC	25 ms; 10,752 KB
testcase_05	AC	27 ms; 10,752 KB
testcase_06	AC	29 ms; 10,624 KB
testcase_07	AC	26 ms; 10,752 KB
testcase_08	AC	25 ms; 10,624 KB
testcase_09	AC	25 ms; 10,752 KB
testcase_10	AC	26 ms; 10,752 KB
testcase_11	AC	303 ms; 10,624 KB
testcase_12	AC	897 ms; 10,752 KB
testcase_13	AC	893 ms; 10,880 KB
testcase_14	WA	-
testcase_15	AC	25 ms; 10,752 KB
testcase_16	AC	25 ms; 10,624 KB
testcase_17	AC	25 ms; 10,624 KB
testcase_18	AC	24 ms; 10,752 KB
testcase_19	AC	367 ms; 10,752 KB
testcase_20	AC	748 ms; 10,752 KB
testcase_21	AC	703 ms; 10,624 KB
testcase_22	AC	716 ms; 10,624 KB
testcase_23	AC	447 ms; 10,624 KB
testcase_24	AC	376 ms; 10,624 KB
testcase_25	WA	-
testcase_26	AC	572 ms; 10,752 KB
testcase_27	AC	773 ms; 10,624 KB
testcase_28	AC	549 ms; 10,752 KB
testcase_29	AC	890 ms; 10,752 KB

ソースコード

import math

def isprime(n):
	if n == 1:
		return False
	else:
		for i in range(2, math.floor(math.sqrt(n))+1):
			if n % i == 0:
				return False
		return True

def bunnkai(n):
	pf = []
	for i in range(2, math.floor(math.sqrt(n))+1):
		if n % i == 0:
			pf.append(i)
			n //= i
	return pf

n = int(input())

if n == 1:
	print("NO")
elif isprime(n):
	print("NO")
elif float.is_integer(math.sqrt(n)):
	if isprime(math.floor(math.sqrt(n))):
		print("NO")
	else:
		print("YES")
else:
	soinsu = bunnkai(n)
	cnt = len(soinsu)
	if cnt == 1 or cnt == 0:
		print("NO")
	else:
		print("YES")