【编程入门题--自定义函数处理素数】

题目描述

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

示例

输入格式

一个数

输出格式

如果是素数输出prime 如果不是输出not prime

样例输入

97

样例输出

prime

解析

素数

• 只有两个正自然数除数-一个和它本身。 质数的相反是合成数
• 0和1既不是素数也不是合数
• 2是质数

代码

def fun(x):
    if x<2:
        t="not prime"
    else:
        for i in range(2,x):
            if x%i==0:
                t="not prime"
                break
            else:
                t="prime"
    return t

if __name__ == '__main__':
    x=int(input())
    print(fun(x))