[编程入门]自定义函数处理素数-题解（python代码）

解题思路:    素数只有1和它本身两个因子，因此可以通过判断是否含有其他因子进而判断是否为素数，利用for循环和%来计算并用计数器count来便于输出。

注意事项:    为减少循环，for到n的平方跟处

参考代码:

n = int(input())
count = 0
for i in range(2, int(pow(n,1/2))+1):
   if n % i == 0:
       count += 1
if count == 0:
   print('prime')
else:
   print('not prime')