我的解题方式（python）

#输入一个数
n=int(input())
#判断你是否为质数
for i in range (2,n):
if n%i==0:
print("No")
break
else:
if n !=1:
print("Yes")
else:
print("no")