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

• 問題: No.36 素数が嫌い！
• ユーザー: neko_the_shadow
• 提出日時: 2016-08-28 00:10:07
• 言語: Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
• 結果: AC
• 実行時間: 1,362 ms / 5,000 ms
• コード長: 280 bytes
• コンパイル時間: 106 ms
• コンパイル使用メモリ: 10,492 KB
• 実行使用メモリ: 8,024 KB
• 最終ジャッジ日時: 2023-09-09 07:41:42
• 合計ジャッジ時間: 17,312 ms
• ジャッジサーバーID （参考情報）: judge15 / judge14

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	628 ms; 7,812 KB
testcase_01	AC	901 ms; 7,880 KB
testcase_02	AC	16 ms; 7,936 KB
testcase_03	AC	16 ms; 7,964 KB
testcase_04	AC	15 ms; 7,916 KB
testcase_05	AC	19 ms; 7,812 KB
testcase_06	AC	20 ms; 8,024 KB
testcase_07	AC	19 ms; 8,016 KB
testcase_08	AC	17 ms; 7,952 KB
testcase_09	AC	17 ms; 7,960 KB
testcase_10	AC	18 ms; 7,908 KB
testcase_11	AC	445 ms; 8,016 KB
testcase_12	AC	1,362 ms; 7,904 KB
testcase_13	AC	1,337 ms; 7,844 KB
testcase_14	AC	660 ms; 7,960 KB
testcase_15	AC	15 ms; 8,016 KB
testcase_16	AC	15 ms; 7,872 KB
testcase_17	AC	16 ms; 7,976 KB
testcase_18	AC	16 ms; 7,896 KB
testcase_19	AC	535 ms; 7,892 KB
testcase_20	AC	1,166 ms; 7,872 KB
testcase_21	AC	1,029 ms; 7,848 KB
testcase_22	AC	1,057 ms; 7,876 KB
testcase_23	AC	649 ms; 7,980 KB
testcase_24	AC	602 ms; 7,840 KB
testcase_25	AC	698 ms; 7,868 KB
testcase_26	AC	829 ms; 7,932 KB
testcase_27	AC	1,148 ms; 7,960 KB
testcase_28	AC	807 ms; 7,884 KB
testcase_29	AC	1,290 ms; 7,844 KB

ソースコード

import math

if __name__ == '__main__':
    n = int(input())

    cnt = 0
    prime = 0
    for prime in range(2, int(math.sqrt(n)) + 1):
        while n % prime == 0:
            cnt += 1
            n //= prime
    
    if n != 1: cnt += 1
    print("YES" if cnt >= 3 else "NO")