#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
• コンパイル時間: 943 ms
• コンパイル使用メモリ: 10,848 KB
• 実行使用メモリ: 8,152 KB
• 最終ジャッジ日時: 2023-09-22 14:39:01
• 合計ジャッジ時間: 10,888 ms
• ジャッジサーバーID （参考情報）: judge14 / judge11

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	AC	473 ms; 8,008 KB
testcase_02	AC	16 ms; 7,984 KB
testcase_03	AC	16 ms; 7,936 KB
testcase_04	AC	16 ms; 7,984 KB
testcase_05	AC	18 ms; 8,060 KB
testcase_06	AC	20 ms; 8,152 KB
testcase_07	AC	19 ms; 7,980 KB
testcase_08	AC	17 ms; 8,012 KB
testcase_09	AC	18 ms; 7,912 KB
testcase_10	AC	17 ms; 7,936 KB
testcase_11	AC	277 ms; 8,152 KB
testcase_12	AC	808 ms; 8,008 KB
testcase_13	AC	806 ms; 8,012 KB
testcase_14	WA	-
testcase_15	AC	16 ms; 7,944 KB
testcase_16	AC	16 ms; 7,980 KB
testcase_17	AC	16 ms; 8,036 KB
testcase_18	AC	15 ms; 7,980 KB
testcase_19	AC	326 ms; 7,956 KB
testcase_20	AC	630 ms; 7,984 KB
testcase_21	AC	614 ms; 8,016 KB
testcase_22	AC	638 ms; 7,984 KB
testcase_23	AC	390 ms; 8,152 KB
testcase_24	AC	329 ms; 7,944 KB
testcase_25	WA	-
testcase_26	AC	496 ms; 7,948 KB
testcase_27	AC	697 ms; 7,988 KB
testcase_28	AC	486 ms; 8,012 KB
testcase_29	AC	782 ms; 7,980 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")