巧妙处理素数问题（python）

解题思路:将因子输入列表中，元素个数仅为2时，判定为素数。

注意事项:

参考代码:

def f(n):
    s=[]
    for i in range(1,n+1):
        if n%i==0:
            s.append(i)
    if len(s)==2:
        print("prime")
    else:
        print("not prime")
def main():
    a=int(input())
    return f(a)
if __name__=='__main__':
    main()