素数判断 python

目录

题目描述

AC代码

题目描述

写一个判断素数的函数，在主函数输入一个整数，输出是否素数的信息。

输入

判断次数和每次输入的任意整数

输出

每次的输入是否为素数

输入样例1 

4
17
5
6
19

输出样例1

prime
prime
not prime
prime

AC代码

import math


def isPrime(x):
    if x < 2:
        print("not prime")
        return False
    for i in range(2, int(math.sqrt(x)) + 1):
        if x % i == 0:
            print("not prime")
            return False
    print("prime")
    return True


t = int(input())
while t > 0:
    t = t - 1
    isPrime(int(input()))