#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
• コンパイル時間: 77 ms
• コンパイル使用メモリ: 12,672 KB
• 実行使用メモリ: 11,008 KB
• 最終ジャッジ日時: 2024-07-08 06:15:51
• 合計ジャッジ時間: 12,209 ms
• ジャッジサーバーID （参考情報）: judge2 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	439 ms; 10,880 KB
testcase_01	AC	544 ms; 10,880 KB
testcase_02	AC	24 ms; 10,752 KB
testcase_03	WA	-
testcase_04	AC	24 ms; 10,880 KB
testcase_05	WA	-
testcase_06	WA	-
testcase_07	AC	27 ms; 10,880 KB
testcase_08	AC	25 ms; 10,752 KB
testcase_09	AC	26 ms; 10,880 KB
testcase_10	AC	26 ms; 10,752 KB
testcase_11	AC	309 ms; 10,752 KB
testcase_12	AC	909 ms; 10,752 KB
testcase_13	AC	895 ms; 10,752 KB
testcase_14	AC	588 ms; 10,752 KB
testcase_15	AC	29 ms; 10,752 KB
testcase_16	AC	25 ms; 10,880 KB
testcase_17	AC	24 ms; 10,752 KB
testcase_18	AC	26 ms; 10,752 KB
testcase_19	AC	377 ms; 10,880 KB
testcase_20	AC	746 ms; 10,880 KB
testcase_21	AC	708 ms; 10,880 KB
testcase_22	AC	734 ms; 10,752 KB
testcase_23	AC	444 ms; 10,752 KB
testcase_24	AC	369 ms; 10,880 KB
testcase_25	AC	483 ms; 10,752 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")