输入形式和输出形式谢谢了

import math

def prime(n):
    if n == 1:
        return False
    tmp = int(math.sqrt(n))
    for i in range(2, tmp+1):
        if n % i == 0:
            return False
    return True

n = int(input())
print(prime(n))