(n: int)
| 30 | |
| 31 | # tag::IS_PRIME[] |
| 32 | def is_prime(n: int) -> bool: |
| 33 | if n < 2: |
| 34 | return False |
| 35 | if n == 2: |
| 36 | return True |
| 37 | if n % 2 == 0: |
| 38 | return False |
| 39 | |
| 40 | root = math.isqrt(n) |
| 41 | for i in range(3, root + 1, 2): |
| 42 | if n % i == 0: |
| 43 | return False |
| 44 | return True |
| 45 | # end::IS_PRIME[] |
| 46 | |
| 47 | if __name__ == '__main__': |