[编程入门]自定义函数处理素数

解题思路:

注意事项:

参考代码:

from math import sqrt

def isPrime(n):

    if n==1:

        return False

    for j in range(2,int(sqrt(n)+1)):

        if n%j==0:

            return False

    return True

def main():

    num=int(input())

    if isPrime(num):

        print("prime")

    else:

        print("not prime")

if __name__=="__main__":

    main()