1057: 素数判定-python

题目描述：

输入一个正整数n，判断n是否是素数，若n是素数，输出”Yes”,否则输出”No”。
输入
输入一个正整数n(n<=1000)
输出
如果n是素数输出"Yes"，否则输出"No"。输出占一行。
样例输入 Copy
2
样例输出 Copy
Yes

答案：

n=int(input())
s=0
for i in range(2,n):
    if n%i==0:
        s=1
        break
if s==1 or n==1:
    print('No')
else:
    print('Yes')