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

• 問題: No.36 素数が嫌い！
• ユーザー: shobonvip
• 提出日時: 2020-05-10 20:34:02
• 言語: Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
• 結果: WA
• 実行時間: -
• コード長: 618 bytes
• コンパイル時間: 97 ms
• コンパイル使用メモリ: 10,908 KB
• 実行使用メモリ: 8,196 KB
• 最終ジャッジ日時: 2023-09-22 14:40:26
• 合計ジャッジ時間: 11,051 ms
• ジャッジサーバーID （参考情報）: judge13 / judge12

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	386 ms; 7,916 KB
testcase_01	AC	475 ms; 8,004 KB
testcase_02	AC	16 ms; 8,048 KB
testcase_03	WA	-
testcase_04	AC	16 ms; 7,968 KB
testcase_05	WA	-
testcase_06	WA	-
testcase_07	AC	17 ms; 8,152 KB
testcase_08	AC	16 ms; 8,124 KB
testcase_09	AC	17 ms; 8,004 KB
testcase_10	AC	17 ms; 8,080 KB
testcase_11	AC	274 ms; 7,968 KB
testcase_12	AC	806 ms; 8,144 KB
testcase_13	AC	804 ms; 8,164 KB
testcase_14	AC	509 ms; 7,960 KB
testcase_15	AC	16 ms; 7,916 KB
testcase_16	AC	16 ms; 8,008 KB
testcase_17	AC	16 ms; 8,008 KB
testcase_18	AC	16 ms; 8,040 KB
testcase_19	AC	328 ms; 8,152 KB
testcase_20	AC	632 ms; 8,052 KB
testcase_21	AC	619 ms; 7,956 KB
testcase_22	AC	640 ms; 7,992 KB
testcase_23	AC	386 ms; 8,008 KB
testcase_24	AC	327 ms; 7,996 KB
testcase_25	AC	428 ms; 8,148 KB
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-

ソースコード

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:
		if n%soinsu[0] == 0:
			print("YES")
		else:
			print("NO")
	else:
		print("YES")