[Python] L1-028. 判断素数-PAT团体程序设计天梯赛GPLT

本题的目标很简单，就是判断一个给定的正整数是否素数。
输入格式：
输入在第一行给出一个正整数N（<=10），随后N行，每行给出一个小于231的需要判断的正整数。
输出格式：
对每个需要判断的正整数，如果它是素数，则在一行中输出“Yes”，否则输出“No”。
输入样例：
2
11
111
输出样例：
Yes
No

PS：感谢github用户@zhuzihao-hz提供的pull request～

import math


def funx(x):
    k = 1
    for i in range(2, int(math.sqrt(x)) + 1):
        if x % i == 0:
            k = 0
    if x == 1:
        k = 0
    return k


N = int(input())
M = []
for i in range(N):
    temp = int(input())
    M.append(funx(temp))
for i in M:
    if i == 1:
        print('Yes')
    else:
        print('No')

import math

def funx(x):

k = 1

for i in range(2, int(math.sqrt(x)) + 1):

if x % i == 0:

k = 0

if x == 1:

k = 0

return k

N = int(input())

M = []

for i in range(N):

temp = int(input())

M.append(funx(temp))

for i in M:

if i == 1:

print('Yes')

else:

print('No')