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

• 問題: No.36 素数が嫌い！
• ユーザー: szkhts
• 提出日時: 2023-04-02 08:46:28
• 言語: Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
• 結果: AC
• 実行時間: 921 ms / 5,000 ms
• コード長: 284 bytes
• コンパイル時間: 314 ms
• コンパイル使用メモリ: 11,784 KB
• 実行使用メモリ: 10,168 KB
• 最終ジャッジ日時: 2023-10-25 00:29:41
• 合計ジャッジ時間: 12,485 ms
• ジャッジサーバーID （参考情報）: judge12 / judge14

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	436 ms; 10,168 KB
testcase_01	AC	559 ms; 10,168 KB
testcase_02	AC	29 ms; 10,168 KB
testcase_03	AC	29 ms; 10,168 KB
testcase_04	AC	29 ms; 10,168 KB
testcase_05	AC	31 ms; 10,168 KB
testcase_06	AC	32 ms; 10,168 KB
testcase_07	AC	31 ms; 10,168 KB
testcase_08	AC	30 ms; 10,168 KB
testcase_09	AC	30 ms; 10,168 KB
testcase_10	AC	30 ms; 10,168 KB
testcase_11	AC	319 ms; 10,168 KB
testcase_12	AC	921 ms; 10,168 KB
testcase_13	AC	917 ms; 10,168 KB
testcase_14	AC	393 ms; 10,168 KB
testcase_15	AC	29 ms; 10,168 KB
testcase_16	AC	28 ms; 10,168 KB
testcase_17	AC	29 ms; 10,168 KB
testcase_18	AC	29 ms; 10,168 KB
testcase_19	AC	372 ms; 10,168 KB
testcase_20	AC	731 ms; 10,168 KB
testcase_21	AC	696 ms; 10,168 KB
testcase_22	AC	717 ms; 10,168 KB
testcase_23	AC	449 ms; 10,168 KB
testcase_24	AC	382 ms; 10,168 KB
testcase_25	AC	480 ms; 10,168 KB
testcase_26	AC	560 ms; 10,168 KB
testcase_27	AC	778 ms; 10,168 KB
testcase_28	AC	549 ms; 10,168 KB
testcase_29	AC	872 ms; 10,168 KB

ソースコード

import math

n = int(input())

def is_prime(n):
    cnt = 0
    for i in range(2, int(math.sqrt(n)) + 1):
        while n % i == 0:
            n //= i
            cnt += 1
    
    if n != 1:
        cnt += 1
    return cnt

if is_prime(n) < 3:
    print("NO")
else:
    print("YES")