题解 1029: [编程入门]自定义函数处理素数

解题思路:

注意事项:

参考代码:

def pri (n):

    if n==1:

      a=('prime')

      return a

    else:

      for i in range(2,n):

         if n%i==0:

            a=('not prime')

            return a

            break

         else:

            a=('prime')

            return a

            break

n=int(input())

print(pri(n))